Update file trainer.cli.py
Browse files- trainer.cli.py +1 -1
trainer.cli.py
CHANGED
|
@@ -43,7 +43,7 @@ if __name__ == '__main__':
|
|
| 43 |
|
| 44 |
|
| 45 |
model = Model(config.model)
|
| 46 |
-
|
| 47 |
|
| 48 |
config.trainer.model = model
|
| 49 |
config.trainer.wandb = wandb
|
|
|
|
| 43 |
|
| 44 |
|
| 45 |
model = Model(config.model)
|
| 46 |
+
wandb = Wandb(config.wandb)
|
| 47 |
|
| 48 |
config.trainer.model = model
|
| 49 |
config.trainer.wandb = wandb
|