def __init__(self, net, cfg, data_iter=None, log_name='logs.txt'): self._input_list = [] self._output_list = [] self._outputs = [] self.graph_ops = None self.net = net self.cfg = cfg self.cur_epoch = 0 self.summary_dict = {} # timer self.tot_timer = Timer() self.gpu_timer = Timer() self.read_timer = Timer() # logger self.logger = colorlogger(cfg.log_dir, log_name=log_name) # initialize tensorflow tfconfig = tf.ConfigProto(allow_soft_placement=True, log_device_placement=False) tfconfig.gpu_options.allow_growth = True self.sess = tf.Session(config=tfconfig) # build_graph self.build_graph() # get data iter self._data_iter = data_iter
def __init__(self, log_name='logs.txt'): self.cur_epoch = 0 # timer self.tot_timer = Timer() self.gpu_timer = Timer() self.read_timer = Timer() # logger self.logger = colorlogger(cfg.log_dir, log_name=log_name)