suluyan.sly 3 years ago
parent
commit
2863a8f7fa
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      modelscope/trainers/hooks/__init__.py

+ 1
- 1
modelscope/trainers/hooks/__init__.py View File

@@ -26,7 +26,7 @@ else:
'iter_timer_hook': ['IterTimerHook'],
'logger': ['TensorboardHook', 'TextLoggerHook'],
'lr_scheduler_hook': ['LrSchedulerHook', 'NoneLrSchedulerHook'],
'optimizer_hook': [
'optimizer': [
'ApexAMPOptimizerHook', 'NoneOptimizerHook', 'OptimizerHook',
'TorchAMPOptimizerHook'
],


Loading…
Cancel
Save