add test config files for torchvision models

obss · Jun 16, 2022 · 11126c2 · 11126c2
1 parent 7464ea3
commit 11126c2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/tests/data/models/torchvision/fasterrcnn_resnet50_fpn.yaml b/tests/data/models/torchvision/fasterrcnn_resnet50_fpn.yaml
@@ -0,0 +1,2 @@
+model_name: fasterrcnn_resnet50_fpn
+num_classes: 91
diff --git a/tests/data/models/torchvision/ssd300_vgg16.yaml b/tests/data/models/torchvision/ssd300_vgg16.yaml
@@ -0,0 +1,2 @@
+model_name: ssd300_vgg16
+num_classes: 91