This commit is contained in:
Starbeamrainbowlabs 2023-01-09 18:25:16 +00:00
parent 581006cbe6
commit 4b7df39fac
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -153,7 +153,7 @@ history = model.fit(dataset_train,
separator="\t"
),
CallbackCustomModelCheckpoint(
model_to_checkpoint=model_predict,
model_to_checkpoint=model,
filepath=os.path.join(
DIR_OUTPUT,
"checkpoints"