FMNIST Data Set

class deepobs.pytorch.datasets.fmnist.fmnist(batch_size, train_eval_size=10000)[source]

DeepOBS data set class for the Fashion-MNIST (FMNIST) 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.