config.json

{"resnet_model": "resnet18", "num_classes": 2}