add dropout layer for tractate

src/models/esmm/model.py

@@ -16,6 +16,7 @@ def build_deep_layer(net, params):
                               units=num_hidden_units,
                               activation=tf.nn.relu,
                               kernel_initializer=tf.glorot_uniform_initializer())
+        net = tf.layers.dropout(net, rate=0.1, training=True)
     return net
 
 

src/train_tractate.py

@@ -73,7 +73,8 @@ def main():
     set_essm_model_save_path("tractate", save_path)
     print("============================================================")
 
-    save_path = get_essm_model_save_path("tractate")
+    # TODO save
+    # save_path = get_essm_model_save_path("tractate")
     print("load path: " + save_path)
 
     # save_path = str(Path("~/data/models/tractate/1598236893").expanduser())  # local