Pytorch学习笔记(九):Pytorch模型的FLOPs、模型参数量等信息输出(torchstat、thop、ptflops、torchsummary)
相关文章Pytorch学习笔记(一):torch.cat()模块的详解Pytorch学习笔记(二):nn.Conv2d()函数详解Pytorch学习笔记(三):nn.BatchNorm2d()函数详解Pytorch学习笔记(四):nn.MaxPool2d()函数详解Pytorch学习笔记(五):nn.AdaptiveAvgPool2d()函数详解Pytorch学习笔记(六):view()和nn.L