From 2d7ae6a99fbfa53cc178a48e1919502c1ad24c92 Mon Sep 17 00:00:00 2001
From: fcakyon <fcakyon@gmail.com>
Date: Thu, 16 Jun 2022 23:07:14 +0300
Subject: [PATCH] add test config files for torchvision models

---
 tests/data/models/torchvision/fasterrcnn_resnet50_fpn.yaml | 2 ++
 tests/data/models/torchvision/ssd300_vgg16.yaml            | 2 ++
 2 files changed, 4 insertions(+)
 create mode 100644 tests/data/models/torchvision/fasterrcnn_resnet50_fpn.yaml
 create mode 100644 tests/data/models/torchvision/ssd300_vgg16.yaml

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 00000000..ec0d0fe4
--- /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 00000000..dfe887c2
--- /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