【转载】https://blog.csdn.net/weixin_41851439/article/details/94644045
使用默认为8的batch size训练:
python train.py --config=yolact_base_config
根据根据实际情况可以适当调节batch size的大小,如果显卡性能低可以减小batch size大小:
python train.py --config=yolact_base_config --batch_size=5
从中断的权重继续训练:
python train.py --config=yolact_base_config --resume=weights/yolact_base_10_32100.pth --start_iter=-1
帮助:
python train.py --help
————————————————
版权声明:本文为CSDN博主「Runtian Qin」的原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/weixin_41851439/article/details/94644045