输入尺寸:
torch.Size([4, 128, 32, 28, 28])
打印网络结构:
print(a.rgb_path.layer2_lateral)
ConvModule(
(conv): Conv3d(128, 128, kernel_size=(7, 1, 1), stride=(4, 1, 1), padding=(3, 0, 0), bias=False)
)
输出尺寸:
torch.Size([4, 128, 8, 28, 28])
输入尺寸:
torch.Size([4, 128, 32, 28, 28])
打印网络结构:
print(a.rgb_path.layer2_lateral)
ConvModule(
(conv): Conv3d(128, 128, kernel_size=(7, 1, 1), stride=(4, 1, 1), padding=(3, 0, 0), bias=False)
)
输出尺寸:
torch.Size([4, 128, 8, 28, 28])