Update file trainer.py
Browse files- trainer.py +1 -1
trainer.py
CHANGED
|
@@ -28,7 +28,7 @@ class Trainer:
|
|
| 28 |
|
| 29 |
for self.epoch in range(self.num_epochs):
|
| 30 |
for self.batch in range(self.num_batches):
|
| 31 |
-
ids = batches[batch]
|
| 32 |
|
| 33 |
loss = self.model.compute_loss(ids)
|
| 34 |
|
|
|
|
| 28 |
|
| 29 |
for self.epoch in range(self.num_epochs):
|
| 30 |
for self.batch in range(self.num_batches):
|
| 31 |
+
ids = batches[self.batch]
|
| 32 |
|
| 33 |
loss = self.model.compute_loss(ids)
|
| 34 |
|