| from torch.utils.data import DataLoader | |
| from data_splitter import train_set, val_set | |
| batch_size = 32 | |
| train_loader = DataLoader(train_set, batch_size, shuffle=True) | |
| val_loader = DataLoader(val_set, batch_size) |
| from torch.utils.data import DataLoader | |
| from data_splitter import train_set, val_set | |
| batch_size = 32 | |
| train_loader = DataLoader(train_set, batch_size, shuffle=True) | |
| val_loader = DataLoader(val_set, batch_size) |