[BETA] Easy-to-use pytorch model trainer, dataloader creators, and utilities + my own implementations of select models
This is a repository of select implementations of deep learning models using pytorch.
There is also my own mini-framework for model training
(which I call simply NetworkTrainer
), that ended up with
something very similar to ignite,
created in order to reduce common boilerplating.
More flexible and intuitive than ignite, in my opinion :).
More models to be added, and improvements on NetworkTrainer
is under way.
pip3 install pytorch-land
See pypi page for package details.
Required packages are specified in requirements.txt file. The packages can be installed using the following command:
pip3 install -r requirements.txt
Notably, the codes are compatible with pytorch 0.4 - working on with pytorch 1.1 compatibility.