Skip to content
Snippets Groups Projects
Commit 4d568393 authored by i-robot's avatar i-robot Committed by Gitee
Browse files

!534 Only set pretrain_ckpt on modelarts when config.pretrain_ckpt is set.

Merge pull request !534 from chenhaozhe/fix-mobilenetv2
parents a435f137 c551c522
No related branches found
No related tags found
No related merge requests found
......@@ -94,7 +94,8 @@ def modelarts_pre_process():
print("#" * 200, os.listdir(os.path.join(config.data_path, config.modelarts_dataset_unzip_name)))
config.dataset_path = os.path.join(config.data_path, config.modelarts_dataset_unzip_name)
config.pretrain_ckpt = os.path.join(config.output_path, config.pretrain_ckpt)
if config.pretrain_ckpt:
config.pretrain_ckpt = os.path.join(config.output_path, config.pretrain_ckpt)
def build_params_groups(net):
decayed_params = []
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment