【模型训练】

【模型训练】

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.

【推广】免费学中医，健康全家人

原文地址：https://www.cnblogs.com/lyj0123/p/13926894.html