opensource/PaddleOCR
3
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2026-09-24 23:33:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fixed bug tto check if log_writer is not None before calling log_model

Browse Source
This commit is contained in:
Manan Goel
2022-04-15 13:43:22 +00:00
parent f161045675
commit eae14caba0
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tools/program.py
+4 -3
View File
@@ -380,7 +380,8 @@ def train(config,
epoch=epoch,
global_step=global_step)
log_writer.log_model(is_best=False, prefix="latest")
if log_writer is not None:
log_writer.log_model(is_best=False, prefix="latest")
if dist.get_rank() == 0 and epoch > 0 and epoch % save_epoch_step == 0:
save_model(
@@ -394,8 +395,8 @@ def train(config,
best_model_dict=best_model_dict,
epoch=epoch,
global_step=global_step)
log_writer.log_model(is_best=False, prefix='iter_epoch_{}'.format(epoch))
if log_writer is not None:
log_writer.log_model(is_best=False, prefix='iter_epoch_{}'.format(epoch))
best_str = 'best metric, {}'.format(', '.join(
['{}: {}'.format(k, v) for k, v in best_model_dict.items()]))