from network import Net as net
model = net(num_channels=1, scale_factor=4, d=32, s=5, m=1)
print(model)
#model = torch.load(pretrained_model, map_location = torch.device('cpu'))
model.load_state_dict(torch.load(pretrained_model, map_location = torch.device('cpu'<
网络学习中参数量具体如何计算
最新推荐文章于 2023-10-30 20:35:32 发布