sir i have this issue in learn.summary() …
Thanks!
You didn’t provide much to go off of, but I see the issue. In the definition for CustomUnet, adjust here:
nf = num_features_model(nn.Sequential(layers[-1]))
And remove that * 2
(will push to git momentarily)
thanks…muellerzr
successfully I solved this error…
thanks…