def init_network(self): # jh-future:sizes can be add into tf.app.flags network = my_model.MyInception(FLAGS.num_classes - FLAGS.labels_offset, [FLAGS.scale_size1, FLAGS.scale_size2], FLAGS.model_name, is_training=False) self.network = network
def init_network(self): # jh-future:sizes can be add into tf.app.flags network = my_model.MyInception( FLAGS.num_classes - FLAGS.labels_offset, [FLAGS.size_branch_0, FLAGS.size_branch_1], FLAGS.model_name, is_training=True) self.network = network