Update file trainer.py

trainer.py

@@ -4,13 +4,13 @@ from util import Config
 
 
 
-class NoOptimizer(torch.optim.Optimizer):
-    def __init__(self, params, lr=0):
-        defaults = dict(lr=lr)
-        super(NoOptimizer, self).__init__(params, defaults)
-
-    def step(self, closure=None):
-        pass
+#class NoOptimizer(torch.optim.Optimizer):
+#    def __init__(self, params, lr=0):
+#        defaults = dict(lr=lr)
+#        super(NoOptimizer, self).__init__(params, defaults)
+#
+#    def step(self, closure=None):
+#        pass
 
 
 
@@ -19,7 +19,6 @@ class Trainer:
         self.__dict__ = dict(config.__dict__)
         
 
-
     def log(self, loss: float):
         print(f"Epoch: {self.epoch} / {self.num_epochs}\t\tBatch: {self.batch} / {self.num_batches}\t\tLoss: {round(loss, 4)}")
         
@@ -32,9 +31,9 @@ class Trainer:
 
 
     def train(self, batches):
-        #self.optimizer = torch.optim.Adam(self.model.parameters(), lr=self.learning_rate)
-        self.optimizer = NoOptimizer(self.model.parameters(), lr=self.learning_rate)
-        
+        self.optimizer = torch.optim.Adam(self.model.parameters(), lr=self.learning_rate)
+        #self.optimizer = NoOptimizer(self.model.parameters(), lr=self.learning_rate)
+
         self.model.unfreeze()
 
         for self.epoch in range(self.num_epochs):