| { | |
| "model_type": "resnet34-lora", | |
| "architecture": "ResNet34 with LoRA adapters", | |
| "task": "image-classification", | |
| "num_classes": 2, | |
| "class_names": ["Non-Smoker", "Smoker"], | |
| "lora_config": { | |
| "rank": 8, | |
| "target_layers": ["layer3", "layer4"] | |
| }, | |
| "input_size": [224, 224], | |
| "pretrained_weights": "ImageNet", | |
| "framework": "PyTorch" | |
| } |