(device, total_epochs, save_every, batch_size)
| 64 | |
| 65 | |
| 66 | def main(device, total_epochs, save_every, batch_size): |
| 67 | dataset, model, optimizer = load_train_objs() |
| 68 | train_data = prepare_dataloader(dataset, batch_size) |
| 69 | trainer = Trainer(model, train_data, optimizer, device, save_every) |
| 70 | trainer.train(total_epochs) |
| 71 | |
| 72 | |
| 73 | if __name__ == "__main__": |
no test coverage detected