def main():
t = time()
conf_code = extract_config_code()
check_flags()
print(get_model_info_as_str())
data = SiameseModelData(FLAGS.dataset_train)
dist_sim_calculator = DistSimCalculator(FLAGS.dataset_train, FLAGS.ds_metric, FLAGS.ds_algo)
model = create_model(FLAGS.model, data.input_dim(), data, dist_sim_calculator)
os.environ["CUDA_VISIBLE_DEVICES"] = str(FLAGS.gpu)
config = tf.compat.v1.ConfigProto()
config.gpu_options.allow_growth = True
sess = tf.compat.v1.Session(config=config)
saver = Saver(sess)
sess.run(tf.compat.v1.global_variables_initializer())
train_costs, train_times = train_loop(data, model, saver, sess)
test(data, model, saver, sess)
saver.save_conf_code(conf_code)
overall_time = convert_long_time_to_str(time() - t)
print(overall_time, saver.get_log_dir())
saver.save_overall_time(overall_time)
return train_costs, train_times