BayesWatch / pytorch-experiments-template

A pytorch based classification experiments template
GNU General Public License v3.0
5 stars 1 forks source link

added a generator to data.random_split to make data splits consistent #39

Closed eleanor91888388 closed 3 years ago

eleanor91888388 commented 3 years ago

I added a generator (manual seed) to data.random_split to make data splits consistent across seeds. Previously the training and validation splits were different for each seed causing data leakage and irreproducibility. Documentation: https://pytorch.org/docs/stable/data.html#torch.utils.data.random_split