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