hi, can you show the config when you were training on the SHB dataset? like lr and optim, etc Or it is just the same with the config in the `config.py`