TylerYep / torchinfo

View model summaries in PyTorch!

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

PANFEI-CHENG opened this issue · comments

commented

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()))

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.