Is not work about 'dict' input?

sksq96 / pytorch-summary

Model summary in PyTorch similar to `model.summary()` in Keras

MIT License

4.01k stars 412 forks source link

Open kdheejb7 opened 3 years ago

kdheejb7 commented 3 years ago

Hello, I'm trying to use pytorch-summary

But, my model's input is "dict" type. What can I do for using this library about dict input?

Thank you for sharing awesome library :)

styanddty commented 3 years ago

I have the same problem. Have you solved it？

Bingoang commented 2 years ago

@kdheejb7 @styanddty Hi, I met the same problem, too. Have you solved it？

kingzone commented 1 year ago

Same problem. Have you solved it?