Attach transforms to model (#9028)

* 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>

classify/val.py

@@ -39,7 +39,7 @@ def run(
     project=ROOT / 'runs/val-cls',  # save to project/name
     name='exp',  # save to project/name
     exist_ok=False,  # existing project/name ok, do not increment
-    half=True,  # use FP16 half-precision inference
+    half=False,  # use FP16 half-precision inference
     dnn=False,  # use OpenCV DNN for ONNX inference
     model=None,
     dataloader=None,
@@ -124,7 +124,6 @@ def run(
         LOGGER.info(f'Speed: %.1fms pre-process, %.1fms inference, %.1fms post-process per image at shape {shape}' % t)
         LOGGER.info(f"Results saved to {colorstr('bold', save_dir)}")
 
-    model.float()  # for training
     return top1, top5, loss