提交 56523aaf 编写于 作者: C chenguowei01

update train.py and utils.py

上级 4db05742
......@@ -121,7 +121,7 @@ def parse_args():
parser.add_argument(
'--use_vdl',
dest='use_vdl',
help='Whether to record the data during training to VisualDL',
help='Whether to record the data to VisualDL during training',
action='store_true')
return parser.parse_args()
......
......@@ -76,6 +76,8 @@ def load_pretrained_model(model, pretrained_model):
raise ValueError(
'The pretrained model directory is not Found: {}'.formnat(
pretrained_model))
else:
logging.info('No pretrained model to load, train from scratch')
def resume(model, optimizer, resume_model):
......@@ -94,6 +96,8 @@ def resume(model, optimizer, resume_model):
raise ValueError(
'The resume model directory is not Found: {}'.formnat(
resume_model))
else:
logging.info('No model need to resume')
def visualize(image, result, save_dir=None, weight=0.6):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册