The new PyTorch Profiler (torch. inside your model, both CPU and GPU, ... I am a network API user and I find that in profiler the reshape layer may cost ...
確定! 回上一頁