【模型训练】
1. 加载预训练模型
1 start_epoch = 0 2 pretrain = True 3 if pretrain: 4 # 加载上一次训练的模型,继续训练 5 start_epoch = 80 6 pretrain_model_path = os.path.join(MODEL_DIR, '预训练权重文件名称') 7 pretrain_model, _ = fluid.dygraph.load_dygraph(pretrain_model_path) 8 model.load_dict(pretrain_model) 9 for epoch in range(start_epoch, TRAIN_EPOCH): 10 # 正常训练过程
2.
3.