Update file trainer.py

trainer.py

@@ -27,10 +27,10 @@ class Trainer:
             for self.batch in range(self.num_batches):
                 ids = batches[self.batch]
 
-                #loss = self.model.compute_loss(ids)
+                loss = self.model.compute_loss(ids)
 
-                #self.optimizer.zero_grad()
-                #loss.backward()
-                #self.optimizer.step()
+                self.optimizer.zero_grad()
+                loss.backward()
+                self.optimizer.step()
 
-                #self.log(loss.item())
+                self.log(loss.item())