Update file trainer.cli.py
Browse files- trainer.cli.py +1 -1
trainer.cli.py
CHANGED
|
@@ -35,7 +35,7 @@ if __name__ == '__main__':
|
|
| 35 |
|
| 36 |
|
| 37 |
batches, num_batches = dataset.batch(ids)
|
| 38 |
-
|
| 39 |
|
| 40 |
print(f"batches: {num_batches}")
|
| 41 |
|
|
|
|
| 35 |
|
| 36 |
|
| 37 |
batches, num_batches = dataset.batch(ids)
|
| 38 |
+
config.trainer.num_batches = num_batches
|
| 39 |
|
| 40 |
print(f"batches: {num_batches}")
|
| 41 |
|