added settings for resnet18-based model

2024-10-19 19:14:34 +00:00
parent a54e7bbbb4
commit ad069269ac
1 changed files with 14 additions and 0 deletions
@@ -0,0 +1,14 @@
+{
+    "datasets": {
+        "train": {
+            "filelist": "datasets/train.csv"
+        },
+        "val": {
+            "filelist": "datasets/val.csv"
+        }
+    },
+    "backbone": {
+        "class": "model.src.models.VisualCommunicationModel",
+        "model_name": "pretrained"
+    }
+}