currently it saves all the checkpoints which use up project storage on wandb too fast. need to only save the best model to the wandb folder.