torch.profiler.profileの機能を使ってプロファイルを取ることができる。プロファイルとは要するにどの関数や処理でどれほどの時間を要しているのかを計測すること。 計測 # profiler setting if config.profile: prof = torch.profiler.profile( activities…