diff --git a/conf/app.yaml b/conf/app.yaml index 3c5cb2c..3d8c04a 100644 --- a/conf/app.yaml +++ b/conf/app.yaml @@ -15,7 +15,7 @@ training: batch_size: 64 epochs: 50 learning_rate: 0.001 - checkpoint_dir: checkpoints + checkpoint_dir: checkpoint_files checkpoint_filename: inference_recognition_model use_gpu: true