Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-25 04:29:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

[to #42322933] fix bug for text logger

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10560149
This commit is contained in:
pangda
2022-10-28 15:26:17 +08:00
committed by yingda.chen
parent 46cfa177aa
commit 303ae2ff36
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modelscope/trainers/hooks/logger/text_logger_hook.py
View File

@@ -61,7 +61,7 @@ class TextLoggerHook(LoggerHook):
self.json_log_path = osp.join(self.out_dir,
'{}.log.json'.format(trainer.timestamp))
if hasattr(trainer, 'meta') and trainer.meta is not None:
self._dump_log(trainer.meta, trainer)
self._dump_log(trainer.meta)
def _get_max_memory(self, trainer):
device = getattr(trainer.model, 'output_device', None)

2
modelscope/trainers/trainer.py
View File

@@ -183,7 +183,7 @@ class EpochBasedTrainer(BaseTrainer):
preprocessor=self.eval_preprocessor,
**kwargs)
self.train_data_collator, self.eval_default_collate = None, None
self.train_data_collator, self.eval_data_collator = None, None
if isinstance(data_collator, Mapping):
if not (ConfigKeys.train in data_collator
or ConfigKeys.val in data_collator):