Update file trainer.py

trainer.py

@@ -24,13 +24,13 @@ class Trainer:
     def train(self, batches): 
         #pass
 
-        model.unfreeze()
+        self.model.unfreeze()
 
         for self.epoch in range(self.num_epochs):
             for self.batch in range(self.num_batches):
                 ids = batches[batch]
 
-                loss = model.compute_loss(ids)
+                loss = self.model.compute_loss(ids)
 
                 self.optimizer.zero_grad()
                 loss.backward()