Assertion error fired

[loisbijing]

I run into assertion error running the profiler in step 4: File "........./lib/python3.6/site-packaegs/apex/pyprof/prof/linear.py", line 87, in init assert (d.name in Linear.biasKernels) AssertionError

I'm using pytorch version 1.10.0+cu102

Please help! Thanks a lot!

[crcrpar]

unfortunately pyprof in apex doesn't seem maintained well and according to https://github.com/NVIDIA/PyProf#pyprof---pytorch-profiling-tool, I guess it's better to use dlprof