Created by: 80917663
问题类型:模型训练
问题描述
模型训练后,使用pdx.load_model接口读取权重文件后报错,无法继续训练。
读取的权重文件是训练中保存的文件,未导出。
模型新建代码:
model = pdx.seg.HRNet(num_classes=num_classes, use_dice_loss=True)
模型读取代码:
model = pdx.load_model('output/hrnet/epoch_200/')
错误原因:
model.trainable = False