* Attach transforms to model
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update val.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update train.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>