Home Explore Help
Sign In
algorithm
/
recommend-model
15
0
Fork 0
Files Issues 0 Pull Requests 0 Wiki
Browse Source

dnn test

丁云鹏 5 months ago
parent
1d3c313173
commit
07c99fa45f
1 changed files with 1 additions and 1 deletions
Split View Show Diff Stats
  1. 1 1
      recommend-model-produce/src/main/python/models/dnn/config.yaml

+ 1 - 1
recommend-model-produce/src/main/python/models/dnn/config.yaml
View File 

@@ -21,7 +21,7 @@ runner:
 
   train_reader_path: "criteo_reader" # importlib format
 
   use_gpu: False
 
   use_auc: False
 
-  train_batch_size: 100
 
+  train_batch_size: 2
 
   epochs: 1
 
   print_interval: 2
 
   model_save_path: "output_model_dnn"