Print Autograd Graph

This is my result for resnet18().
I can not find any problem. @zym1010

2 Likes