I am studying the tutorial given here to profile the CPU time of the model. I get an error message ModuleNotFoundError: No module named 'torch.profiler'. I have torch version 1.7.1 installed. May I know what’s the problem?
The new profiler was released in 1.8.1 as described in this blog post so you would have to update PyTorch.
@ptrblck It worked. Thanks!