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