Используйте torchsummary на Pytorch
NoSuchKey
рекомендация
отblog.csdn.net/Doraemon_Zzn/article/details/104456627
рекомендация
ранжирование