How to train large models (that take a lot of time)?