Fix missing `load_only_model_parameters` parameter in the entrypoint train function

Jay Chanrequested to merge
jay_fix_entry_point into dev

Fix the bug that was introduced in !111 (merged)

Merge request reports