train
mode007 opened this issue · comments
预训练train.py保存的是.weights文件，finetune用的是.ckpt文件，请问这个是怎么转化过去的呢？
预训练train.py保存的是.weights文件，finetune用的是.ckpt文件，请问这个是怎么转化过去的呢？
请问老哥学会怎么转了吗？finetune之后怎么重新加载进行预测呢
同问，我修改了finetune为多卡机制，保存后的模型文件是
best_model.weights.data-00000-of-00002，best_model.weights.data-00001-of-00002，best_model.weights.data.index，checkpoint
不知道该怎么加载，用load_weights_from_checkpoint会报错