madhavkarthi's picture
Upload config.yaml with huggingface_hub
50fb30c verified
architecture: ResNet18
classes:
- cafe
- gym
- library
- outdoor
input_size:
- 224
- 224
num_classes: 4
pretrained: ImageNet
task: image-classification