if __name__ == '__main__': res18 = ResNet(block=BasicBlock, n_blocks=[2, 2, 2, 2], channels=[64, 128, 256, 512], output_dim=1000) input = torch.randn(1, 3, 224, 224) output = res18(input) print(output.size())
时间: 2023-05-12 15:03:20 浏览: 64
ResNet代码
这是一个使用 PyTorch 实现的 ResNet 网络,包含了 4 个阶段,每个阶段包含了 2 个 BasicBlock 模块,输入图片的大小为 224x224,输出维度为 1000。在输入一个 3 通道的 224x224 的随机张量后,输出的大小应该为 (1, 1000)。
阅读全文