Opening Pytorch Summary… Continue