Created by: fxwfzsxyq

用动态图参照torchvision model修改了一个vgg，训练发现卷积，bn，relu之后的feature map中出现inf，不知道在那个位置计算出现了inf，导致loss为nan