fix: training

[othertea]

After the recently merged PRs, training with usual configs no longer worked. (I'm not sure how others have been running training scripts; I did not see any new training configs.)
This PR makes minimal changes so that we can once again train with the two provided configs: protein_lm/configs/train/toy_localcsv.yaml, protein_lm/configs/train/toy_hf.yaml.

[jamaliki]

LGTM!

[NZ99]

This looks good to me too -- thank you very much @othertea!

[talkhanz]

Hey @othertea.

I'm running into issues when do_curriculum_learning is true.
The fields removed from protein_lm/configs/train/toy_localcsv.yaml and protein_lm/configs/train/toy_hfyaml as per this issue are required for curriculum learning to take place.

Can you let me know if you can reproduce an error when do_curriculum_learning is true in the config file?

[othertea]

Hi @talkhanz ,
Please feel free to create a new yaml file that does curriculum learning with the desired specifications! As their names indicate, the yamls in the repo are "toy" yamls that are used for sanity checking, not yamls intended to be used for the full training workflows. In particular, they are not guaranteed to work when you modify a single value, such as setting do_curriculum_learning to true.
This PR updated the yamls because the previous versions of the yamls did not run.

[talkhanz]

makes sense. Thankyou @othertea!