【模型训练】

 

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.

posted @ 2020-11-04 17:13  Bro_Li  阅读(164)  评论(0编辑  收藏  举报