()
| 59 | GPU_MEM_FRACTION = 0.8 |
| 60 | |
| 61 | def log_dir(): |
| 62 | log_dir = FLAGS.base_log_dir + "/unsup-" + FLAGS.train_prefix.split("/")[-2] |
| 63 | log_dir += "/{model:s}_{model_size:s}_{lr:0.6f}/".format( |
| 64 | model=FLAGS.model, |
| 65 | model_size=FLAGS.model_size, |
| 66 | lr=FLAGS.learning_rate) |
| 67 | if not os.path.exists(log_dir): |
| 68 | os.makedirs(log_dir) |
| 69 | return log_dir |
| 70 | |
| 71 | # Define model evaluation function |
| 72 | def evaluate(sess, model, minibatch_iter, size=None): |