()
| 48 | |
| 49 | |
| 50 | def load_train_objs(): |
| 51 | train_set = MyTrainDataset(2048) # load your dataset |
| 52 | model = torch.nn.Linear(20, 1) # load your model |
| 53 | optimizer = torch.optim.SGD(model.parameters(), lr=1e-3) |
| 54 | return train_set, model, optimizer |
| 55 | |
| 56 | |
| 57 | def prepare_dataloader(dataset: Dataset, batch_size: int): |