1、使用torchsummary来打印网络特征提取部分参数实例:import torch from torchsummary import summary from model import AlexNet device=torch.device("cuda" if torch.cuda.is_available() else "cpu" ) net=AlexNet().to(device)
Copyright © 2005-2024 51CTO.COM 版权所有 京ICP证060544号