- A simple implementation of classes allowing supervised learning on images.
- There are two popular architectures, Inception and ResNet developed from scratch and an implementation of auxiliary classifiers and a "Binary Inception loss".
- You can also find classes for preparation and training allowing tracking with tensorboard as well as various advanced manipulations such as hook functions.
- Everything has been tested
-
Notifications
You must be signed in to change notification settings - Fork 0
ilanaliouchouche/PyTorchCVTrainingPack
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Personal pack I use as reference for Computer Vision projects