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

[PANFEI-CHENG]

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]

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.