summary

PHOTO EMBED

Tue Nov 29 2022 23:22:32 GMT+0000 (Coordinated Universal Time)

Saved by @olyaTh #pytorch #summary #model

from torchsummary import summary
summ = summary(model, torch.Size((1,28,28)))
content_copyCOPY