Skip to content

Commit 30db3f3

Browse files
Add missing ops for ResNet-50 (#16)
* Add Conv support * Add poolings and reshape * Add Cargo.lock * Add Identity * Add ResNet-50 conversion test * cargo fmt --------- Co-authored-by: Stephan Seitz <sseitz@nvidia.com>
1 parent 4dd1ec6 commit 30db3f3

11 files changed

Lines changed: 3276 additions & 10 deletions

File tree

.github/workflows/ci.yml

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,15 @@ jobs:
6666
restore-keys: |
6767
${{ runner.os }}-target-
6868
69+
# Cache the ResNet-50 ONNX model separately so it survives Cargo.lock churn
70+
# and is not evicted as collateral damage when the target/ cache is invalidated.
71+
# Static key — bump the suffix only if the upstream model URL changes.
72+
- name: Cache ResNet-50 ONNX model
73+
uses: actions/cache@v4
74+
with:
75+
path: target/test-data
76+
key: resnet50-onnx-opset16-v1
77+
6978
- name: Check formatting
7079
run: make fmt-check
7180

0 commit comments

Comments
 (0)