Fix checkpoint path

zhiqwang · Nov 9, 2020 · e5424cf · e5424cf
1 parent 960a64c
commit e5424cf
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 13 deletions.
diff --git a/.github/workflows/nightly.yml b/.github/workflows/nightly.yml
@@ -43,11 +43,8 @@ jobs:
         cmake .. -DTorch_DIR=$TORCH_PATH/share/cmake/Torch
         make -j4
         sudo make install
-    - name: Download pretrained checkpoints and generate torchscript
+    - name: Export torchscript model
       run: |
-        cd test/tracing
-        wget https://github.com/zhiqwang/yolov5-rt-stack/releases/download/v0.2.0/yolov5s.pt
-        cd ../..
         python -m test.tracing.trace_model
     - name: Test libtorch tracing
       run: |

diff --git a/.github/workflows/stable.yml b/.github/workflows/stable.yml
@@ -43,11 +43,8 @@ jobs:
         cmake .. -DTorch_DIR=$TORCH_PATH/share/cmake/Torch
         make -j4
         sudo make install
-    - name: Download pretrained checkpoints and generate torchscript
+    - name: Export torchscript model
       run: |
-        cd test/tracing
-        wget https://github.com/zhiqwang/yolov5-rt-stack/releases/download/v0.2.0/yolov5s.pt
-        cd ../..
         python -m test.tracing.trace_model
     - name: Test libtorch tracing
       run: |

diff --git a/models/yolo.py b/models/yolo.py
@@ -139,7 +139,7 @@ def forward(
 
 model_urls = {
     'yolov5s':
-        'https://github.com/zhiqwang/yolov5-rt-stack/releases/download/v0.2.0/yolov5s.pt',
+        'https://github.com/zhiqwang/yolov5-rt-stack/releases/download/v0.2.1/yolov5s.pt',
 }
 
 

diff --git a/test/tracing/trace_model.py b/test/tracing/trace_model.py
@@ -5,10 +5,7 @@
 
 if __name__ == "__main__":
 
-    model = yolov5(
-        cfg_path='./models/yolov5s.yaml',
-        checkpoint_path='./test/tracing/yolov5s.pt',
-    )
+    model = yolov5(pretrained=True)
     model.eval()
 
     traced_model = torch.jit.script(model)