如何优雅的阅读DBNet的训练代码
程序员文章站
2022-08-18 08:14:43
工程链接:https://github.com/MhLiao/DB最近在学习DBNet,然后由于本人想偷懒,然后使用了以上链接的工程(这样就不需要升级cuda和cudnn),最后发现该工程实在是写的太优雅了,完全不知道哪里是数据加载,哪里是测试,各种数据增强又在什么位置,所以现在就我的经验给大家盘点一下;遇到报错balance_cross_entropy_loss.py", line 49, in forward negative_loss, _ = torch.topk(negativ...
工程链接:https://github.com/MhLiao/DB
最近在学习DBNet,然后由于本人想偷懒,然后使用了以上链接的工程(这样就不需要升级cuda和cudnn),最后发现该工程实在是写的太优雅了,完全不知道哪里是数据加载,哪里是测试,各种数据增强又在什么位置,所以现在就我的经验给大家盘点一下;
目录
遇到报错:failed to synchronize: device-side assert triggered
本文地址:https://blog.csdn.net/qq_22764813/article/details/107430550
上一篇: js实现移动端图片滑块验证功能