Unable to load parameters initial_training.pt during training

[jonathanking]

I've found that I'm unable to load the provided openfold model parameters when using --resume_model_weights_only=True in train_openfold.py . An error arises because the keys corresponding the model parameters are prefixed using model.*, while the training code expects the prefix module.*. I have a small modification in my branch to support this.

Is the training script supposed to support loading the provided openfold parameters? If not, is this a useful feature of interest to others? Happy to contribute my modification.