* add callbacks to train function in wandb sweep

Fix following https://github.com/ultralytics/yolov5/pull/4688 which modified the function signature to `train`

* Cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>