Add Hub custom models to CI tests (#4978)

* Update ci-testing.yml for Hub custom model tests

* Update ci-testing.yml

.github/workflows/ci-testing.yml

@@ -65,19 +65,25 @@ jobs:
       - name: Tests workflow
         run: |
           # export PYTHONPATH="$PWD"  # to run '$ python *.py' files in subdirectories
-          di=cpu # inference devices  # define device
+          di=cpu  # device
 
-          # train
+          # Train
           python train.py --img 128 --batch 16 --weights ${{ matrix.model }}.pt --cfg ${{ matrix.model }}.yaml --epochs 1 --device $di
-          # detect
+          # Val
-          python detect.py --weights ${{ matrix.model }}.pt --device $di
-          python detect.py --weights runs/train/exp/weights/last.pt --device $di
-          # val
           python val.py --img 128 --batch 16 --weights ${{ matrix.model }}.pt --device $di
           python val.py --img 128 --batch 16 --weights runs/train/exp/weights/last.pt --device $di
-
+          # Detect
+          python detect.py --weights ${{ matrix.model }}.pt --device $di
+          python detect.py --weights runs/train/exp/weights/last.pt --device $di
           python hubconf.py  # hub
+          # Export
           python models/yolo.py --cfg ${{ matrix.model }}.yaml  # build PyTorch model
           python models/tf.py --weights ${{ matrix.model }}.pt  # build TensorFlow model
           python export.py --img 128 --batch 1 --weights ${{ matrix.model }}.pt --include torchscript onnx  # export
+          # Python
+          python - <<EOF
+          import torch
+          model = torch.hub.load('ultralytics/yolov5', 'custom', path='runs/train/exp/weights/last.pt')
+          EOF
+
         shell: bash