Replace replace calls to model.train() for executorch models

Replace replace calls to model.train() for executorch models #1686