From 091b56fcd34899cb17b52d8ae6ef8fb9b5fde00a Mon Sep 17 00:00:00 2001 From: ancestor-mithril Date: Wed, 8 May 2024 11:17:22 +0300 Subject: [PATCH] Fixed bug with poping relevant arguments out of args --- utils/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/trainer.py b/utils/trainer.py index 709fbb4..6e67e1b 100644 --- a/utils/trainer.py +++ b/utils/trainer.py @@ -64,7 +64,7 @@ def es_metric(self): return 'Train/Loss' if not hasattr(self.args, 'es_metric') else self.args.es_metric def init_logdir(self): - params = self.args.__dict__ + params = {**self.args.__dict__} params.pop('device') params.pop('data_path') params.pop('disable_progress_bar')