度量学习finetune阶段训练出错
Created by: yanmeizhao
paddle版本1.8,目前已经完成了训练阶段,准备开始finetune,但是发现finetune无法训起来,finetune脚本:
python train_pair.py
--model=ResNet50
--train_batch_size=160
--test_batch_size=50
--lr=0.0001
--total_iter_num=100000
--use_gpu=True
--pretrained_model=model_save/ResNet50/30000
--model_save_dir=finetune_output
--loss_name=eml
--samples_each_class=2
其中model_save/ResNet50/30000是第一阶段训练好的模型,测试其Recall为0.78174,但是启动finetune脚本后,出现如下错误: 麻烦帮忙看一下