Expose wandb.init argument resume to neural-lam.train_model

[sadamov]

Wandb.init has a resume flag where you can continue existing experiments based on id: https://docs.wandb.ai/ref/python/init/. I find that quite useful, especially on HPC systems that crash often or have limited runtime for jobs.

This should require changes only in train_model and in utils.