Skip to content

Latest commit

 

History

History
13 lines (9 loc) · 562 Bytes

File metadata and controls

13 lines (9 loc) · 562 Bytes

定义统一模型接口后的训练 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 方向的位置信息。