Update 2 files

- /trainer.cli.py
- /trainer.py

trainer.cli.py

@@ -40,5 +40,11 @@ if __name__ == '__main__':
     print(f"batches: {num_batches}")
 
 
-    trainer = Trainer(config)
-    trainer.train(dataset)
+    model = Model(config.model)
+    self.wandb = Wandb(config.wandb)
+
+    config.trainer.model = model
+    config.trainer.wandb = wandb
+
+    trainer = Trainer(config.trainer)
+    trainer.train(batches)

trainer.py

@@ -8,12 +8,7 @@ from model import Model
 class Trainer:
     def __init__(self, config: Config):
         self.__dict__ = dict(config.trainer.__dict__)
-
-        
-        #self.wandb = Wandb(config.wandb)
-        self.model = Model(config.model)
         
-        self.optimizer = torch.optim.Adam(self.model.parameters(), lr=self.learning_rate)
 
 
     def log(self, loss: float):
@@ -21,9 +16,8 @@ class Trainer:
         
 
 
-    def train(self, batches): 
-        #pass
-
+    def train(self, batches):
+        self.optimizer = torch.optim.Adam(self.model.parameters(), lr=self.learning_rate)
         self.model.unfreeze()
 
         for self.epoch in range(self.num_epochs):