n0obcoder · November 21, 2019 18:01
diff --git a/mobile_gallery_pytorch_01_preprocessing_and_dataloaders_2.py b/mobile_gallery_pytorch_01_preprocessing_and_dataloaders_2.py
 # Defining dataloaders which would return data in batches
 batch_size = 64
 train_loader = torch.utils.data.DataLoader(train_dataset, batch_size = batch_size, shuffle = True)
 val_loader   = torch.utils.data.DataLoader(val_dataset,   batch_size = batch_size, shuffle = False)

 print('number of batches in train_loader with a batch_size of {}: {}'.format(batch_size,  len(train_loader)))
 print('number of batches in val_loader with a batch_size of {}: {}'.format(batch_size,  len(val_loader)))
diff --git a/mobile_gallery_pytorch_01_preprocessing_and_dataloaders_2_output.py b/mobile_gallery_pytorch_01_preprocessing_and_dataloaders_2_output.py
 number of batches in train_loader with a batch_size of 64: 16
 number of batches in val_loader with a batch_size of 64: 4
	# Defining dataloaders which would return data in batches
	batch_size = 64
	train_loader = torch.utils.data.DataLoader(train_dataset, batch_size = batch_size, shuffle = True)
	val_loader = torch.utils.data.DataLoader(val_dataset, batch_size = batch_size, shuffle = False)

	print('number of batches in train_loader with a batch_size of {}: {}'.format(batch_size, len(train_loader)))
	print('number of batches in val_loader with a batch_size of {}: {}'.format(batch_size, len(val_loader)))
	number of batches in train_loader with a batch_size of 64: 16
	number of batches in val_loader with a batch_size of 64: 4