def __init__(self, model, fn_A, fn_B, batch_size=64): self.batch_size = batch_size self.model = model self.images_A = minibatchAB(fn_A, self.batch_size) self.images_B = minibatchAB(fn_B, self.batch_size)
def __init__(self, model, fn_A, fn_B): self.model = model self.images_A = minibatchAB(fn_A, self.BATCH_SIZE) self.images_B = minibatchAB(fn_B, self.BATCH_SIZE)