定义统一模型接口后的训练 Pipeline 示例:
model = SimpleCNN().to("cuda") # ✅ 任意 BaseModel 子类
train_loader = DataLoader(dataset, batch_size=32, shuffle=True)
train_model(model, train_loader, epochs=10, lr=1e-3)```
改进方案:
- 需要提升数据维度,现在是一个 1D 超声信号中包含了缺陷位置信息。
- 实际情况是一个二维板面的振动信息包含了缺陷位置信息,所以信号需要提升到三维,包含 X 方向和 Y 方向的位置信息。