Skip to main content

Return to Answer

Post Timeline

deleted 1 character in body
Source Link
Eric Aya
Eric Aya
  • 70.2k
  • 36
  • 191
  • 266

Keras like model summary using torchsummary

Keras like model summary using torchsummary:

from torchsummary import summary
summary(model, input_size=(3, 224, 224))

Keras like model summary using torchsummary

from torchsummary import summary
summary(model, input_size=(3, 224, 224))

Keras like model summary using torchsummary:

from torchsummary import summary
summary(model, input_size=(3, 224, 224))
Source Link
Abhishek Singh
Abhishek Singh
  • 123
  • 1
  • 4

Keras like model summary using torchsummary

from torchsummary import summary
summary(model, input_size=(3, 224, 224))