pangda yingda.chen 3 years ago
parent
commit
303ae2ff36
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      modelscope/trainers/hooks/logger/text_logger_hook.py
  2. +1
    -1
      modelscope/trainers/trainer.py

+ 1
- 1
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, self.json_log_path = osp.join(self.out_dir,
'{}.log.json'.format(trainer.timestamp)) '{}.log.json'.format(trainer.timestamp))
if hasattr(trainer, 'meta') and trainer.meta is not None: 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): def _get_max_memory(self, trainer):
device = getattr(trainer.model, 'output_device', None) device = getattr(trainer.model, 'output_device', None)


+ 1
- 1
modelscope/trainers/trainer.py View File

@@ -183,7 +183,7 @@ class EpochBasedTrainer(BaseTrainer):
preprocessor=self.eval_preprocessor, preprocessor=self.eval_preprocessor,
**kwargs) **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 isinstance(data_collator, Mapping):
if not (ConfigKeys.train in data_collator if not (ConfigKeys.train in data_collator
or ConfigKeys.val in data_collator): or ConfigKeys.val in data_collator):


Loading…
Cancel
Save