By Bokundev High Quality - Training Slayer V740

# Load dataset and create data loader dataset = MyDataset(data, labels) data_loader = DataLoader(dataset, batch_size=batch_size, shuffle=True)