def get_loaders()

in touch_charts/recon.py [0:0]

• 10 lines of code
• 2 McCabe index (conditional complexity)

	def get_loaders(self):
		# training data
		train_data = data_loaders.mesh_loader_touch(self.classes, self.args, set_type='train')
		train_loader = DataLoader(train_data, batch_size=self.args.batch_size, shuffle=True, num_workers=16, collate_fn=train_data.collate)

		# validation data
		valid_loaders = []
		set_type = 'test' if self.args.eval else 'valid'
		for c in self.classes:
			valid_data = data_loaders.mesh_loader_touch(c, self.args, set_type=set_type)
			valid_loaders.append(
				DataLoader(valid_data, batch_size=self.args.batch_size, shuffle=False, num_workers=16, collate_fn=valid_data.collate))
		return train_loader, valid_loaders