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