CMT(
(stem_conv1): Conv2d(3, 16, kernel_size=(3, 3), stride=(2, 2), padding=(1, 1))
(stem_relu1): GELU()
(stem_norm1): BatchNorm2d(16, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
(stem_conv2): Conv2d(16, 16, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
(stem_relu2): GELU()
(stem_norm2): BatchNorm2d(16, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
(stem_conv3): Conv2d(16, 16, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
(stem_relu3): GELU()
(stem_norm3): BatchNorm2d(16, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
(patch_embed_a): PatchEmbed(
(proj): Conv2d(16, 46, kernel_size=(2, 2), stride=(2, 2))
(norm): LayerNorm((46,), eps=1e-05, elementwise_affine=True)
)
(patch_embed_b): PatchEmbed(
(pr
CMT的架构和么一层的网络 CMt cnn mlp transfomer
最新推荐文章于 2023-07-28 17:42:01 发布