Update file trainer.py

trainer.py

@@ -18,7 +18,7 @@ class Trainer:
 
     def train(self, batches):
         self.optimizer = torch.optim.Adam(self.model.parameters(), lr=self.learning_rate)
-        #self.model.unfreeze()
+        self.model.unfreeze()
 
         for self.epoch in range(self.num_epochs):
             for self.batch in range(self.num_batches):
@@ -30,4 +30,4 @@ class Trainer:
                 loss.backward()
                 self.optimizer.step()
 
-                self.log(loss.item())
+                #self.log(loss.item())