krshrimali
Transfer-Learning-Dogs-Cats-Libtorch
Jupyter Notebook

Transfer Learning on Dogs vs Cats dataset using PyTorch C+ API

Last updated Oct 4, 2025
12
Stars
3
Forks
3
Issues
0
Stars/day
Attention Score
3
Language breakdown
Jupyter Notebook 66.8%
C++ 27.1%
Python 4.9%
CMake 1.1%
Files click to expand
README

Transfer-Learning-Dogs-Cats-Libtorch

Transfer Learning on Dogs vs Cats dataset using PyTorch C++ API

Implementation

  • mkdir build
  • cmake -DCMAKEPREFIXPATH=/absolute/path/to/libtorch ..
  • make
  • ./example <pathtoscripting_model>
TODOs:
  • Load dataset in the way suggested. Prevents OOM (lazily load a single image)
  • Test accuracy. And predictions samples.
🔗 More in this category

© 2026 GitRepoTrend · krshrimali/Transfer-Learning-Dogs-Cats-Libtorch · Updated daily from GitHub