unsupported operand type(s) for +: 'int' and 'str'

TylerYep / torchinfo

View model summaries in PyTorch!

MIT License

2.48k stars 117 forks source link

unsupported operand type(s) for +: 'int' and 'str' #268

Open PANFEI-CHENG opened 1 year ago

PANFEI-CHENG commented 1 year ago

when I use torchinfo with model taking dictionary as input summary(self.model, input_data = [data_dict]) I meet a bug ： TypeError (note: full exception trace is shown but execution is paused at: _run_module_as_main) unsupported operand type(s) for +: 'int' and 'str' python3.8/site-packages/torchinfo/torchinfo.py", line 512, in lambda data: (lambda d: sum(d.values()))

snimu commented 11 months ago

Could you please provide the whole exception trace? I cannot debug this with that little data. It would also help if you could share how your data_dict looks exactly, and preferably also how your model is structured.