model-loads (pypi): Loads GPU or CPU pytorch models
Keywords: neural-network|pytorch