architecture: ResNet18
classes:
- cafe
- gym
- library
- outdoor
input_size:
- 224
- 224
num_classes: 4
pretrained: ImageNet
task: image-classification