Update 2 files

- /trainer.py
- /model.py

model.py

@@ -20,16 +20,9 @@ class Model:
         print(f"Model's embedding size is {self.params.vocab_size}")
 
 
-    def parameters(self):
-        return self.model.parameters()
-
-
-    def unfreeze(self):
-        self.model.train()
-
-
-    def freeze(self):
-        self.model.eval()
+    def parameters(self): return self.model.parameters()
+    def unfreeze(self):   self.model.train()
+    def freeze(self):     self.model.eval()
 
 
     def compute_loss(self, input_ids, labels=None, criterion=None):

trainer.py

@@ -30,4 +30,4 @@ class Trainer:
                 loss.backward()
                 self.optimizer.step()
 
-                #self.log(loss.item())
+                self.log(loss.item())