adityaiitb / pyprof2

PyProf2: PyTorch Profiling tool
BSD 3-Clause "New" or "Revised" License
83 stars 11 forks source link

Please document how to add a user layer marker #6

Closed jlquinn closed 4 years ago

jlquinn commented 4 years ago

I'm trying to figure it out from looking at the code, but it would be useful to have a description.

adityaiitb commented 4 years ago

Please look at the description and a resnet50 example here on how to add a user layer marker. Please let me know if the example is clear and you are able to make progress.