修改配置

conf/sample_config.json

@@ -22,7 +22,7 @@
   "remote_url": "https://app.singlewindow.cn/ceb2pubweb//verifyCode/creator",
   "cycle_stop": 6000,
   "acc_stop": 0.99,
-  "cycle_save": 20,
+  "cycle_save": 300,
   "enable_gpu": 0,
   "train_batch_size": 180,
   "test_batch_size": 150

train_model.py

@@ -71,7 +71,7 @@ class TrainModel(CNN):
 
         # test model input and output
         logger.debug(">>> Start model test")
-        batch_x, batch_y = self.get_batch(0, size=10)
+        batch_x, batch_y = self.get_batch(0, size=100)
         logger.debug(">>> input batch images shape: %s", batch_x.shape)
         logger.debug(">>> input batch labels shape: %s", batch_y.shape)