3月終わりごろ、PyTorch の1.8.1がリリースされました。1.8.0から1.8.1でsmall bugfixだけかと思ってたら、元々あったautograd profilerの次のバージョンのprofilerという形でPyTorch Profilerが追加されたことがPyTorchのblogで紹介されました。今回はこのPyTorch Profilerがどんなことができるのか?を確かめてみたので、それについて紹介していきます。 PyTorch Profilerとは? 元々PyTorchにはautograd profiler (torch.autograd.profiler)というprofilerがありました。これを改良してものがPyTorch Profiler (torch.profiler)としてPyTorch 1.8.1で追加されました。blogの記事を読んだり、実際に触ってみた感
![PyTorch 1.8.1で入ったPyTorch Profilerを使ってみた - まったり勉強ノート](https://cdn-ak-scissors.b.st-hatena.com/image/square/3b3be2029853f43fe41472378d55169536ce21c9/height=288;version=1;width=512/https%3A%2F%2Fwww.mattari-benkyo-note.com%2Fwp-content%2Fuploads%2F2021%2F04%2Ftb_overview-1024x313.png)