WIP

main.py

@@ -5,6 +5,12 @@ import dataset
 import models
 
 
+# config = tf.ConfigProto(log_device_placement=True)
+# config.gpu_options.per_process_gpu_memory_fraction = 0.5
+# config.gpu_options.allow_growth = True
+# session = tf.Session(config=config)
+
+
 def main():
     # parameter
     innerCNNFilters = 512