flopml
/

mamba

flpelerin commited on Aug 24, 2024

Commit

920fe2d

1 Parent(s): a740705

Update file trainer.py

Files changed (1) hide show

trainer.py CHANGED Viewed

@@ -11,6 +11,6 @@ class Trainer:
         #self.wandb = Wandb(config.wandb)
         print(f"model config: {config.model}")
-        print(f"config.params: {config.params}")
         #self.model = Model(config.model)

         #self.wandb = Wandb(config.wandb)
         print(f"model config: {config.model}")
+        print(f"config.params: {config.model.params}")
         #self.model = Model(config.model)