Ver Fonte

dnn test

丁云鹏 há 5 meses atrás
pai
commit
07c99fa45f

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

@@ -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"