deepJDOT
deepJDOT copied to clipboard
model.train_on_batch prevents overfitting!
Hi! Thank you for the procedure, I recently applied your program to solve some practical problems. But I encountered some difficulties. In your jdot_align model, you use self.model.train_on_batch([data], [np.vstack((ys, l_dummy)), g_dummy]) to train, but I encountered the problem of overfitting. I learned by checking the information that the function train_on_batch doesn't have Callback functions, So I can't save the best model. Do you have any other good solutions? thanks!