diff --git a/tests/data/models/torchvision/fasterrcnn_resnet50_fpn.yaml b/tests/data/models/torchvision/fasterrcnn_resnet50_fpn.yaml
new file mode 100644
index 000000000..ec0d0fe4b
--- /dev/null
+++ b/tests/data/models/torchvision/fasterrcnn_resnet50_fpn.yaml
@@ -0,0 +1,2 @@
+model_name: fasterrcnn_resnet50_fpn
+num_classes: 91
\ No newline at end of file
diff --git a/tests/data/models/torchvision/ssd300_vgg16.yaml b/tests/data/models/torchvision/ssd300_vgg16.yaml
new file mode 100644
index 000000000..dfe887c27
--- /dev/null
+++ b/tests/data/models/torchvision/ssd300_vgg16.yaml
@@ -0,0 +1,2 @@
+model_name: ssd300_vgg16
+num_classes: 91
\ No newline at end of file