This commit is contained in:
parent
c2a71605f8
commit
34d2431bf0
|
@ -15,7 +15,7 @@ training:
|
||||||
batch_size: 64
|
batch_size: 64
|
||||||
epochs: 50
|
epochs: 50
|
||||||
learning_rate: 0.001
|
learning_rate: 0.001
|
||||||
checkpoint_dir: checkpoints
|
checkpoint_dir: checkpoint_files
|
||||||
checkpoint_filename: inference_recognition_model
|
checkpoint_filename: inference_recognition_model
|
||||||
use_gpu: true
|
use_gpu: true
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user