added loading and testing the model again

This commit is contained in:
2026-05-05 22:42:27 +02:00
parent 7531afcbd7
commit cfa4da697e
4 changed files with 66 additions and 18 deletions

View File

@@ -36,11 +36,15 @@ Root_dir/
```
once this is manually done, We can run finetunning a selected model. For now, 4 models are implemented.
once this is manually done, We can run finetunning a selected model. For now, 8 models are implemented.
- ResNet-18
- ResNet-50
- DenseNet121
- Inception
- GoogleNet
- ShuffleNet
- EfficientNet
- WideResNet
## Fine tuning
### Preparation