Change validation loss instead training loss in corrections

model save callback > name of the loss should be "val_loss" instead "loss"

To upload designs, you'll need to enable LFS and have an admin enable hashed storage. More information