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"
    }
 }