Open zhangnan12d opened 2 years ago
训练的时候遇到了问题,很抱歉打扰您,实在不知道该怎么办
检查一下你的kd_one_stage.py里的 KnowledgeDistillationSingleStageDetector 有没有这一行
感谢感谢!已经可以训练了!但是不知道为什么训练出来的效果特别差(我用的是自定义的SAR船舶检测数据集),是因为epoch太少的缘故吗?希望您能给我一点建议
给出训练完整log
20220702_094100.log 辛苦您了
你的数据集是旋转检测框还是水平框?应该是数据集格式有问题(如果是我我会先检查这个)。其次是注意学习率与GPU数量要匹配一下
你先用gfl训一个试试,因为我看到你教师模型是用的coco上预训练的
20220703_153520.log 感谢两位老师的建议,对我帮助很大。解决方法:1、改用官方的数据集转换脚本,把voc格式转换为coco格式。2、会报loss=nan的错,将默认的学习率0.02改为0.001就可以了(我只有一个gpu)3、导入合适的教师模型