/PyTorch-AbstractionLevels

Five ways of writing a simple artificial neural network in PyTorch with different levels of abstraction. The lowest level only uses a tensor parameter and the highest uses pre defined loss, optimiser and sequential architecture.

Primary LanguagePython

Watchers