torchinfo.summary() is not able to executable

[Weijie-ZHAO]

[Weijie-ZHAO]

Hi, thanks for your liabrary, but could you pls solve this problem when we run torchinfo.summary() to check each layer

Best!

[wavefrontshaping]

Hi,

Are you sure you specified the correct dtype? You should use complex64. It works for me with something like

torchinfo.summary(model, input_size=(1, 1, 28, 28), dtypes = [torch.complex64])