Weights and Biases was used for tracking training performance and model settings. A change is needed to gather_args to ensure model arguments are logged to weights and biases
gather_args
Learner.gather_args
Learner.gather_args()
Gather config parameters accessible to the learner, adds model config logging on top of default wandb logging