-
Notifications
You must be signed in to change notification settings - Fork 6
/
Copy pathtrain.py
27 lines (20 loc) · 824 Bytes
/
train.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
import tensorflow as tf
from datetime import datetime
import model
import data_tf
tf.logging.set_verbosity(tf.logging.INFO)
print(tf.__version__)
time_start = datetime.utcnow()
print("Experiment started at {}".format(time_start.strftime("%H:%M:%S")))
print(".......................................")
model.est.train(data_tf.train_input_fn)
time_end = datetime.utcnow()
print(".......................................")
print("Experiment finished at {}".format(time_end.strftime("%H:%M:%S")))
print("")
time_elapsed = time_end - time_start
print("Experiment elapsed time: {} seconds".format(time_elapsed.total_seconds()))
print("#######################################")
print("############# evaluation ##############")
print("#######################################")
valid = model.est.evaluate(data_tf.eval_input_fn)