yangbofun/pytorch-ops-scripts

transfer aten::ops called in c++ to torch ops called in python

Introduction

This subject is for operators statistics when train models by using pytorch.

Usage

1. Pytorch profiler to Get aten operators by using pytorch profiler. For detials see https://pytorch.org/tutorials/recipes/recipes/profiler_recipe.html?highlight=profiler for detail
2. Modify scripts.sh in order to suit for your profiler files path.
3. Run scripts.sh