Results
1
comments of
Yu Feng
> Using SummaryX solved my probelm Very helpful answer! It works nicely now! https://github.com/nmhkahn/torchsummaryX