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