pytorch DatasetFolder 划分 train and test

totall_set = DatasetFolder("../labeled", loader=tifffile.imread, extensions="tif", transform=None)

train_set, test_set = torch.utils.data.random_split(totall_set, [batch_size_train, batch_size_test])

train_loader = DataLoader(train_set, batch_size=batch_size_train, shuffle=True)
test_loader = DataLoader(test_set, batch_size=batch_size_test, shuffle=True)