From 8fd337b2be89a16bfc324eb3b19ec44db8dac655 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Thu, 18 Aug 2022 21:14:21 +0200
Subject: [PATCH] Add PyTorch Hub classification CI checks

Add PyTorch Hub loading of official and custom trained classification models to CI checks.

May help resolve https://github.com/ultralytics/yolov5/issues/8790#issuecomment-1219840718

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
---
 .github/workflows/ci-testing.yml | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/.github/workflows/ci-testing.yml b/.github/workflows/ci-testing.yml
index aa797c44d487..fde6fffe92f4 100644
--- a/.github/workflows/ci-testing.yml
+++ b/.github/workflows/ci-testing.yml
@@ -133,3 +133,8 @@ jobs:
           python classify/predict.py --imgsz 32 --weights $b --source ../datasets/mnist2560/test/7/60.png  # predict
           python classify/predict.py --imgsz 32 --weights $m --source data/images/bus.jpg  # predict
           python export.py --weights $b --img 64 --imgsz 224 --include torchscript  # export
+          python - <<EOF
+          import torch
+          for path in '$m', '$b':
+              model = torch.hub.load('.', 'custom', path=path, source='local')
+          EOF