remove tensorboard hook as default

tensorboard has been removed from the requirements of framework.txt, so we remove tensorboard hook from default config
        Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11519980

    * remove tensorboard hook as default

* Merge branch 'master' into fix/remove_default_tensorboard_hook
This commit is contained in:
wenmeng.zwm
2023-02-08 10:07:07 +00:00
committed by mulin.lyh
parent c8d4e755fd
commit d5ae8ae43b
2 changed files with 12 additions and 17 deletions

View File

@@ -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'
}]
}
}

View File

@@ -268,6 +268,9 @@ class TrainerTest(unittest.TestCase):
}, {
'type': 'EvaluationHook',
'interval': 1
}, {
'type': 'TensorboardHook',
'interval': 1
}]
},
'evaluation': {