I personally intend to add the TRAIN, but to be honest I didn’t apply it before, however here in the following posts you could find more discussions about this matter,
and this post,
I suggest in this case to rerun the normal stamps, steps without any change, if this technically goes normally without any error, then, I’d suggest to change the parameters,
Here you could find good example it helps too much,