pytorch-basic-module. A wrapped basic model class for pytorch models. You can Inherit it to make your model easier to use. It contains methods such as load, save, multi-thread save, parallel distribution, train, validate, predict and so on.

github.com/miracleyoo/pytorch-basic-module

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

No recent activity.