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)
