PacktPublishing / Modern-Computer-Vision-with-PyTorch

Modern Computer Vision with PyTorch, published by Packt

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

torchsummary has been renamed to torchinfo

jemzipx opened this issue · comments

All lines contained torchsummary throughout the book has to be renamed to torchinfo, i.e.:

from torchinfo import summary

Furthermore, all lines containing summary have to edited with the new syntax:
summary(model, (32,3,64,64))

Library is still backwards compatible so there's no need to change any code