pytorch_model_wrapper
