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.