MDL中finetune阶段的baseline_aic_finetune.yml配置问题

[chufall]

你好！

run.sh前面两步： python train.py --config_file='configs/baseline_aic.yml' OUTPUT_DIR "('/data/model/0402_5/')" python train.py --config_file='configs/baseline_aic_finetune.yml' OUTPUT_DIR "('/data/model/0402_6/')"

我理解: 第一步是进行pre-train阶段，使用imagenet预训练模型，输出resnet101_ibn_a_80.pth 第二步是finetune阶段，使用resnet101_ibn_a_80.pth进行训练，冻结前两层，输出resnet101_ibn_a_40.pth

但是在baseline_aic_finetune.yml里，配置项 MODEL: PRETRAIN_CHOICE: 'finetune' PRETRAIN_PATH: '/data/model/0402_6/resnet101_ibn_a_40.pth'

这个PRETRAIN_PATH是不是写错了，应该是/0402_5/resnet101_ibn_a_80.pth？ 否则在运行这个train之前，这个文件压根没有生成过？

谢谢！

[heshuting555]

你说的是对的呢，当时整理代码着急没注意到呢～～～ 感谢提醒！！！ 谢谢关注～