Dataloader's split - nlp - PyTorch Forums
https://discuss.pytorch.org/t/dataloaders-split/13571102.11.2021 · for that instead of doing it with the dataloader you have to do it beforehand # scikit learns train test split from sklearn.model_selection import train_test_split # I assume you are having some x , y formatted labels X_train , y_train = train_test_split(X,y) # here you can also give out percentage of train and test split # you have default as 80% of data for train and 20% of …