Home

Welcome to the Custom-Model-Training-PyTorch wiki!

PyTorch is an open-source machine learning library for Python, based on Torch, used for applications such as natural language processing. It is primarily developed by Facebook's artificial-intelligence research group. This is an attempt on creating a Custom Model Trainer built on top of PyTorch. The Trainer is capable of preparing models based on Custom Data. It uses a Transfer Learning feature that increases the performance and decreases the computation. At current the trainer is in its initial development, we are open for pull requests and contributions.

We request, for PR please put up your contributions in Issues first.