Skip to content

Commit d429ad6

Browse files
committed
skip config write when not logging to wandb (mainly for Pretrained)
1 parent c25858a commit d429ad6

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

metamon/rl/metamon_to_amago.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -683,6 +683,10 @@ class MetamonAMAGOExperiment(amago.Experiment):
683683
def start(self):
684684
super().start()
685685

686+
def init_logger(self):
687+
if self.log_to_wandb:
688+
super().init_logger()
689+
686690
def init_envs(self):
687691
out = super().init_envs()
688692
amago.utils.call_async_env(self.val_envs, "take_long_break")

0 commit comments

Comments
 (0)