Factorize helper function for all model loading

[sagadre]

It would be nice to simplify the checkpoint loading. Right now it is a bit confusing with checkpoint_path, args.checkpoint_path, join(.., "checkpoints"), remote-sync, etc. all referenced in various places inmain.py