/pytorch_modules

A neural network toolkit built on pytorch/opencv/numpy that includes neural network layers, modules, loss functions, optimizers, data loaders, data augmentation, etc.

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

Watchers