https://github.com/krshrimali/transfer-learning-dogs-cats-libtorch

Transfer Learning on Dogs vs Cats dataset using PyTorch C+ API
https://github.com/krshrimali/transfer-learning-dogs-cats-libtorch

cling dogs-vs-cats jupyter libtorch pytorch pytorch-cpp transfer-learning xeus-cling

Last synced: 2 months ago
JSON representation

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

• Host: GitHub
• URL: https://github.com/krshrimali/transfer-learning-dogs-cats-libtorch
• Owner: krshrimali
• Created: 2019-08-12T07:41:27.000Z (over 5 years ago)
• Default Branch: master
• Last Pushed: 2019-08-16T18:29:36.000Z (over 5 years ago)
• Last Synced: 2024-11-02T14:42:03.510Z (2 months ago)
• Topics: cling, dogs-vs-cats, jupyter, libtorch, pytorch, pytorch-cpp, transfer-learning, xeus-cling
• Language: Jupyter Notebook
• Homepage: https://krshrimali.github.io/Applying-Transfer-Learning-Dogs-Cats/
• Size: 446 KB
• Stars: 13
• Watchers: 4
• Forks: 3
• Open Issues: 3
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# Transfer-Learning-Dogs-Cats-Libtorch

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

**Implementation**

1. `mkdir build`

2. `cmake -DCMAKE_PREFIX_PATH=/absolute/path/to/libtorch ..`

3. `make`

4. `./example `

TODOs:

1. Load dataset in the way suggested. Prevents OOM (lazily load a single image)

2. ~~Test accuracy. And predictions samples.~~