def load_samples(loader,cuda,batch_size):
sample_array = [next(loader) for i in range(0,batch_size)]
batched_data, batched_targets = Batcher.batch_samples(sample_array)
if cuda:
batched_data = [x.cuda() for x in batched_data]
batched_targets = [x.cuda() for x in batched_targets]
return batched_data,batched_targets,sample_array
def load_samples(loader, cuda, batch_size):
sample_array = [loader.next() for i in range(0, batch_size)]
batched_data, batched_targets = Batcher.batch_samples(sample_array)
if cuda:
batched_data = map(lambda x: x.cuda(), batched_data)
batched_targets = map(lambda x: x.cuda(), batched_targets)
return batched_data, batched_targets, sample_array