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

update train.py and utils.py

上级 4db05742
...@@ -121,7 +121,7 @@ def parse_args(): ...@@ -121,7 +121,7 @@ def parse_args():
parser.add_argument( parser.add_argument(
'--use_vdl', '--use_vdl',
dest='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') action='store_true')
return parser.parse_args() return parser.parse_args()
......
...@@ -76,6 +76,8 @@ def load_pretrained_model(model, pretrained_model): ...@@ -76,6 +76,8 @@ def load_pretrained_model(model, pretrained_model):
raise ValueError( raise ValueError(
'The pretrained model directory is not Found: {}'.formnat( 'The pretrained model directory is not Found: {}'.formnat(
pretrained_model)) pretrained_model))
else:
logging.info('No pretrained model to load, train from scratch')
def resume(model, optimizer, resume_model): def resume(model, optimizer, resume_model):
...@@ -94,6 +96,8 @@ def resume(model, optimizer, resume_model): ...@@ -94,6 +96,8 @@ def resume(model, optimizer, resume_model):
raise ValueError( raise ValueError(
'The resume model directory is not Found: {}'.formnat( 'The resume model directory is not Found: {}'.formnat(
resume_model)) resume_model))
else:
logging.info('No model need to resume')
def visualize(image, result, save_dir=None, weight=0.6): 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.
先完成此消息的编辑!
想要评论请 注册