- 博客(3)
- 收藏
- 关注
RSS订阅原创 Deepfake-FFDI-图像赛题 baseline（3）加载模型 部分 @Datawhale AI 夏令营
criterion使用交叉熵损失作为损失函数，并将其移动到 GPU 上（使用.cuda()optimizer使用 Adam 优化器，学习率设置为 0.005。scheduler使用学习率调度器StepLR，每 4 个周期将学习率乘以 0.85。
2024-07-14 22:48:57
490
原创 Deepfake-FFDI-图像赛题 baseline（2）模型训练与验证 部分 @Datawhale AI 夏令营
这段代码定义了两个类和，它们主要用于在训练或验证神经网络模型时，方便地记录和显示性能指标（例如损失、精度等）的平均值和进度。
2024-07-14 22:37:04
642
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人