TylerYep/torchinfo

View model summaries in PyTorch!

语言：Python · GitHub 星标：2,951 · Fork：145

话题：keras · python · pytorch · torch · torch-summary · torchinfo · torchsummary · torchvision

在 GitHub 上查看原仓库