From ad069269ac8fabd8be2ffcb480a45dcd7b6f1f0d Mon Sep 17 00:00:00 2001 From: brian Date: Sat, 19 Oct 2024 19:14:34 +0000 Subject: [PATCH] added settings for resnet18-based model --- model/src/configs/resnet18.json | 14 ++++++++++++++ 1 file changed, 14 insertions(+) create mode 100644 model/src/configs/resnet18.json diff --git a/model/src/configs/resnet18.json b/model/src/configs/resnet18.json new file mode 100644 index 0000000..f21d76a --- /dev/null +++ b/model/src/configs/resnet18.json @@ -0,0 +1,14 @@ +{ + "datasets": { + "train": { + "filelist": "datasets/train.csv" + }, + "val": { + "filelist": "datasets/val.csv" + } + }, + "backbone": { + "class": "model.src.models.VisualCommunicationModel", + "model_name": "pretrained" + } +}