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