摘要: I 验证网络结构是否有误 初始化一个符合网络的输入数据 input = torch.ones((64, 3, 32, 32)) 将输入数据传进网络,看是否报错 print(network(input).shape) II 修改已知网络(比如vgg16) vgg16_false = torchvisi 阅读全文
posted @ 2023-02-01 09:56 Kurie 阅读(150) 评论(0) 推荐(0) 编辑