TylerYep / torchinfo

View model summaries in PyTorch!

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Write/Read Operation counter

qmpzzpmq opened this issue · comments

Is possible add a "write bandwidth", "read bandwidth" into summary function's col_names with specific input.

the net inference performance also depond on not only the computation opertaion also the memory operation as well

just like "MemRead" "MemWrite" in torchstat