pytorch_cute.py

dataset, val_set = torch.utils.data.random_split(dataset,[int(len(dataset) * 0.85), 
                                                 len(dataset) - int(len(dataset) * 0.85)])

config.manual_seed = 33
print("Random Seed: ", config.manual_seed)
random.seed(config.manual_seed)
torch.manual_seed(config.manual_seed)

if torch.cuda.is_available():
    torch.cuda.manual_seed_all(config.manual_seed)

torch.backends.cudnn.deterministic = True
torch.backends.cudnn.benchmark = False

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")