ceykmc/pytorch_model_summary

pytorch model summary, statistic parameters number, memory usage, FLOPs and so on

语言：Python · GitHub 星标：97 · Fork：15

在 GitHub 上查看原仓库