MNIST Data Set¶
-
class
deepobs.pytorch.datasets.mnist.
mnist
(batch_size, train_eval_size=10000)[source]¶ DeepOBS data set class for the MNIST data set.
Parameters: - batch_size (int) -- The mini-batch size to use. Note that, if
batch_size
is not a divider of the dataset size (60 000
for train,10 000
for test) the remainder is dropped in each epoch (after shuffling). - train_eval_size (int) -- Size of the train eval data set.
Defaults to
10 000
the size of the test set.
-
_make_dataloader
()¶ A helper that is shared by all three data loader methods.
- batch_size (int) -- The mini-batch size to use. Note that, if