Man1103/ResNet18-FT-PetImages-11.7M

License & Attribution

This repository contains weights or code derived from the ResNet18 foundational architecture developed by the ResNet Authors. This ResNet18 model is fine-tuned using popular CatsVsDogs PetImages dataset.

• License: Distributed under the https://opensource.org/license/bsd-3-clause.
• Original Model Base: https://huggingface.co/microsoft/resnet-18
• Copyright Notice: Copyright 2025-2026 Hugging Face & The ResNet Authors.
• Research paper introducing this model: https://arxiv.org/abs/1512.03385

Dataset:

You can find the fine-tuning dataset using the link: https://huggingface.co/datasets/Man1103/PetImages.

Fine-tuning Configuration Metrics:

• Num epochs: 30
• Device: CUDA (Nvidia Tesla T4 GPU)
• Training batch size: 128
• Learning rate: 0.001
• Accuracy Score: 0.9330 (93.30%)

Please refer to preprocessing and training pipelines in the model files:

• Preprocessing pipeline: preprocessing.py
• Training pipeline: train.py