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