diff --git a/modelscope/trainers/default_config.py b/modelscope/trainers/default_config.py index 7b2e339a..a02478b9 100644 --- a/modelscope/trainers/default_config.py +++ b/modelscope/trainers/default_config.py @@ -4,23 +4,15 @@ from modelscope.utils.config import Config DEFAULT_CONFIG = { 'train': { - 'hooks': [ - { - 'type': 'CheckpointHook', - 'interval': 1 - }, - { - 'type': 'TextLoggerHook', - 'interval': 10 - }, - { - 'type': 'IterTimerHook' - }, - { - 'type': 'TensorboardHook', - 'interval': 10 - }, - ] + 'hooks': [{ + 'type': 'CheckpointHook', + 'interval': 1 + }, { + 'type': 'TextLoggerHook', + 'interval': 10 + }, { + 'type': 'IterTimerHook' + }] } } diff --git a/tests/trainers/test_trainer.py b/tests/trainers/test_trainer.py index c692196a..4453d4dc 100644 --- a/tests/trainers/test_trainer.py +++ b/tests/trainers/test_trainer.py @@ -268,6 +268,9 @@ class TrainerTest(unittest.TestCase): }, { 'type': 'EvaluationHook', 'interval': 1 + }, { + 'type': 'TensorboardHook', + 'interval': 1 }] }, 'evaluation': {