train.py 第146行训练时候self.__training: False应该设置为True吧

Stinky-Tofu / Stronger-yolo

🔥Improve yolo with latest paper

MIT License

3 stars 0 forks source link

Open zhangming8 opened 5 years ago

zhangming8 commented 5 years ago

非常感谢您的工程，注释很详细。我发现了一个小问题，train.py 第146行训练时候self.__training: False应该设置为True吧？

Stinky-Tofu commented 5 years ago

非常感谢您的工程，注释很详细。我发现了一个小问题，train.py 第146行训练时候self.__training: False应该设置为True吧？

因为我们加载且冻住了所有非检测层，如果训练BN层的话，会使得卷积层的权重不适用，所以我就没有训练BN层，当然我也试着训练过BN层，性能只有一点点下降