This commit is contained in:
parent
2aa5ca3fc1
commit
b4d0ffa58f
|
@ -119,7 +119,7 @@ def train_model():
|
||||||
|
|
||||||
checkpoint_callback = ModelCheckpoint(
|
checkpoint_callback = ModelCheckpoint(
|
||||||
dirpath="checkpoint_files",
|
dirpath="checkpoint_files",
|
||||||
filename="inference_recognition_model.ckpt",
|
filename="inference_recognition_model",
|
||||||
save_top_k=1,
|
save_top_k=1,
|
||||||
verbose=True, # Disable checkpoint verbose output
|
verbose=True, # Disable checkpoint verbose output
|
||||||
monitor="val_acc",
|
monitor="val_acc",
|
||||||
|
|
Loading…
Reference in New Issue
Block a user