目录
3.model.train()与model.eval()的用法
1.pytorch中参数初始化方法
https://www.cnblogs.com/lindaxin/p/8037561.html
yolo_training.py中的weights_init函数采用以下方法对整个网络进行初始化,如图所示:
2.pytorch保存与加载模型来测试或继续训练
https://blog.csdn.net/yanghao201607030101/article/details/110947689
train.py中加载预训练模型进行继续训练,如图所示: