()
| 76 | |
| 77 | |
| 78 | def load_train_objs(): |
| 79 | train_set = MyTrainDataset(2048) # load your dataset |
| 80 | model = torch.nn.Linear(20, 1) # load your model |
| 81 | optimizer = torch.optim.SGD(model.parameters(), lr=1e-3) |
| 82 | return train_set, model, optimizer |
| 83 | |
| 84 | |
| 85 | def prepare_dataloader(dataset: Dataset, batch_size: int): |