Uploading complete config dict to wandb #1
Description
Currrently the model type is added to the config dict inside the model class, and then saved. This is necessary for saving/loading, but currently not included in the config that is pushed to wandb. We also add things like the save url and number of parameters, after using the config to initialise the model. Having seperate instances of the config dict containing different information is a bit annoying, would be nice to find the most concise way to have everything pushe to wandb with minimal lines of code in the training script.