search

sksq96 / pytorch-summary

Model summary in PyTorch similar to `model.summary()` in Keras
MIT License
4.01k stars 412 forks source link

unable to handle multiple inputs #112

Open lzhengchun opened 4 years ago

lzhengchun commented 4 years ago

it does not seem to work when my forward function has more than one args.

TangChiaHsin commented 4 years ago

like this format! summary(net, [(1, 128, 128), (1, 128, 128)])

If it doesn't work, plz show more code and error description.

lzhengchun commented 4 years ago

like this format! summary(net, [(1, 128, 128), (1, 128, 128)])

If it doesn't work, plz show more code and error description.

it works!

Thanks