def __init__(self, opts): self.opts = opts self.batchnum = 1 self.final_reducer = decode_final_reducer(opts.final_reducer) self.build_network() self.train_problems_loader = None
def __init__(self, opts): self.opts = opts self.final_reducer = decode_final_reducer(opts.final_reducer) self.build_network() self.train_problems_loader = None self.train_counter = 0 self.epoch_size = opts.epoch_size self.epoch = 0