[ru999gol]

I don't see much of a reason you want to clutter your code with that, if you need a wrapper around pytorch use fastai not some obscure library nobody has ever heard about.

[wodenokoto]

Thank god you weren’t around when keras was released.

[jsinai]

Use fastai if you want your code to be unreadable, unmaintable and prone to mysterious bugs.