Update file trainer.py

trainer.py

@@ -13,7 +13,7 @@ class Trainer:
         #self.wandb = Wandb(config.wandb)
         self.model = Model(config.model)
         
-        self.optimizer = torch.optim.Adam(model.parameters(), lr=self.learning_rate)
+        self.optimizer = torch.optim.Adam(self.model.parameters(), lr=self.learning_rate)
 
 
     def log(self, loss: float):