FishmanTheMurloc
|
4787f4f2f9
|
尝试复现RodNet的训练结果 (#1688)
* 增加RodNet必须支持训练的单元测试；RodNet的torch链路改造成全张量计算，由此把之前忽略的参数dz、h_coeff、offset变得可学习
* 了解到损失函数CrossEntropyLoss内置了softmax，因此从forward方法中移出；offset是手动指定的偏置值，不是学习得到的，因此也移出到单独的PostProcess方法中
* 根据得到的源码整理RodNet；新增在数据集上达到一定准确率的单元测试
|
2025-06-14 20:03:14 +08:00 |
|