def __init__(self, problem_number, nb_samples, batch_size, cuda = False):
if nb_samples%batch_size > 0:
- print('nb_samples must be a mutiple of batch_size')
+ print('nb_samples must be a multiple of batch_size')
raise
self.cuda = cuda
def __init__(self, problem_number, nb_samples, batch_size, cuda = False):
if nb_samples%batch_size > 0:
- print('nb_samples must be a mutiple of batch_size')
+ print('nb_samples must be a multiple of batch_size')
raise
self.cuda = cuda