cerebras.modelzoo.fc_mnist.pytorch.data.get_train_dataloader#

cerebras.modelzoo.fc_mnist.pytorch.data.get_train_dataloader(params)[source]#
Parameters

params (<dict>) – dict containing input parameters for creating dataset.

Expects the following fields:

  • “data_dir” (string): path to the data files to use.

  • “batch_size” (int): batch size

  • “to_float16” (bool): whether to convert to float16 or not

  • “drop_last_batch” (bool): whether to drop the last batch or not