()
| 67 | |
| 68 | |
| 69 | def load_train_objs(): |
| 70 | train_set = MyTrainDataset(2048) # load your dataset |
| 71 | model = torch.nn.Linear(20, 1) # load your model |
| 72 | optimizer = torch.optim.SGD(model.parameters(), lr=1e-3) |
| 73 | return train_set, model, optimizer |
| 74 | |
| 75 | |
| 76 | def prepare_dataloader(dataset: Dataset, batch_size: int): |