efficienthrnetbc-2的结构

长度是338。

['conv_stem.0.weight:torch.Size([24, 3, 3, 3])', 'conv_stem.1.weight:torch.Size([24])', 'conv_stem.1.bias:torch.Size([24])', 'conv_stem.1.running_mean:torch.Size([24])', 'conv_stem.1.running_var:torch.Size([24])', 'conv_stem.1.num_batches_tracked:torch.Size([])', 'blocks.0.0.conv_dw.0.weight:torch.Size([24, 1, 3, 3])', 'blocks.0.0.conv_dw.1.weight:torch.Size([24])', 'blocks.0.0.conv_dw.1.bias:torch.Size([24])', 'blocks.0.0.conv_dw.1.running_mean:torch.Size([24])', 'blocks.0.0.conv_dw.1.running_var:torch.Size([24])', 'blocks.0.0.conv_dw.1.num_batches_tracked:torch.Size([])', 'blocks.0.0.se.conv_reduce.weight:torch.Size([6, 24, 1, 1])', 'blocks.0.0.se.conv_reduce.bias:torch.Size([6])', 'blocks.0.0.se.conv_expand.weight:torch.Size([24, 6, 1, 1])', 'blocks.0.0.se.conv_expand.bias:torch.Size([24])', 'blocks.0.0.conv_pw.0.weight:torch.Size([16, 24, 1, 1])', 'blocks.0.0.conv_pw.1.weight:torch.Size([16])', 'blocks.0.0.conv_pw.1.bias:torch.Size([16])', 'blocks.0.0.conv_pw.1.running_mean:torch.Size([16])', 'blocks.0.0.conv_pw.1.running_var:torch.Size([16])', 'blocks.0.0.conv_pw.1.num_batches_tracked:torch.Size([])', 'blocks.1.0.conv_pw.0.weight:torch.Size([96, 16, 1, 1])', 'blocks.1.0.conv_pw.1.weight:torch.Size([96])', 'blocks.1.0.conv_pw.1.bias:torch.Size([96])', 'blocks.1.0.conv_pw.1.running_mean:torch.Size([96])', 'blocks.1.0.conv_pw.1.running_var:torch.Size([96])', 'blocks.1.0.conv_pw.1.num_batches_tracked:torch.Size([])', 'blocks.1.0.conv_dw.0.weight:torch.Size([96, 1, 3, 3])', 'blocks.1.0.conv_dw.1.weight:torch.Size([96])', 'blocks.1.0.conv_dw.1.bias:torch.Size([96])', 'blocks.1.0.conv_dw.1.running_mean:torch.Size([96])', 'blocks.1.0.conv_dw.1.running_var:torch.Size([96])', 'blocks.1.0.conv_dw.1.num_batches_tracked:torch.Size([])', 'blocks.1.0.se.conv_reduce.weight:torch.Size([4, 96, 1, 1])', 'blocks.1.0.se.conv_reduce.bias:torch.Size([4])', 'blocks.1.0.se.conv_expand.weight:torch.Size([96, 4, 1, 1])', 'blocks.1.0.se.conv_expand.bias:torch.Size([96])', 'blocks.1.0.conv_pwl.0.weight:torch.Size([24, 96, 1, 1])', 'blocks.1.0.conv_pwl.1.weight:torch.Size([24])', 'blocks.1.0.conv_pwl.1.bias:torch.Size([24])', 'blocks.1.0.conv_pwl.1.running_mean:torch.Size([24])', 'blocks.1.0.conv_pwl.1.running_var:torch.Size([24])', 'blocks.1.0.conv_pwl.1.num_batches_tracked:torch.Size([])', 'blocks.1.1.conv_pw.0.weight:torch.Size([144, 24, 1, 1])', 'blocks.1.1.conv_pw.1.weight:torch.Size([144])', 'blocks.1.1.conv_pw.1.bias:torch.Size([144])', 'blocks.1.1.conv_pw.1.running_mean:torch.Size([144])', 'blocks.1.1.conv_pw.1.running_var:torch.Size([144])', 'blocks.1.1.conv_pw.1.num_batches_tracked:torch.Size([])', 'blocks.1.1.conv_dw.0.weight:torch.Size([144, 1, 3, 3])', 'blocks.1.1.conv_dw.1.weight:torch.Size([144])', 'blocks.1.1.conv_dw.1.bias:torch.Size([144])', 'blocks.1.1.conv_dw.1.running_mean:torch.Size([144])', 'blocks.1.1.conv_dw.1.running_var:torch.Size([144])', 'blocks.1.1.conv_dw.1.num_batches_tracked:torch.Size([])', 'blocks.1.1.se.conv_reduce.weight:torch.Size([6, 144, 1, 1])', 'blocks.1.1.se.conv_reduce.bias:torch.Size([6])', 'blocks.1.1.se.conv_expand.weight:torch.Size([144, 6, 1, 1])', 'blocks.1.1.se.conv_expand.bias:torch.Size([144])', 'blocks.1.1.conv_pwl.0.weight:torch.Size([24, 144, 1, 1])', 'blocks.1.1.conv_pwl.1.weight:torch.Size([24])', 'blocks.1.1.conv_pwl.1.bias:torch.Size([24])', 'blocks.1.1.conv_pwl.1.running_mean:torch.Size([24])', 'blocks.1.1.conv_pwl.1.running_var:torch.Size([24])', 'blocks.1.1.conv_pwl.1.num_batches_tracked:torch.Size([])', 'blocks.2.0.conv_pw.0.weight:torch.Size([144, 24, 1, 1])', 'blocks.2.0.conv_pw.1.weight:torch.Size([144])', 'blocks.2.0.conv_pw.1.bias:torch.Size([144])', 'blocks.2.0.conv_pw.1.running_mean:torch.Size([144])', 'blocks.2.0.conv_pw.1.running_var:torch.Size([144])', 'blocks.2.0.conv_pw.1.num_batches_tracked:torch.Size([])', 'blocks.2.0.conv_dw.0.weight:torch.Size([144, 1, 5, 5])', 'blocks.2.0.conv_dw.1.weight:torch.Size([144])', 'blocks.2.0.conv_dw.1.bias:torch.Size([144])', 'blocks.2.0.conv_dw.1.running_mean:torch.Size([144])', 'blocks.2.0.conv_dw.1.running_var:torch.Size([144])', 'blocks.2.0.conv_dw.1.num_batches_tracked:torch.Size([])', 'blocks.2.0.se.conv_reduce.weight:torch.Size([6, 144, 1, 1])', 'blocks.2.0.se.conv_reduce.bias:torch.Size([6])', 'blocks.2.0.se.conv_expand.weight:torch.Size([144, 6, 1, 1])', 'blocks.2.0.se.conv_expand.bias:torch.Size([144])', 'blocks.2.0.conv_pwl.0.weight:torch.Size([32, 144, 1, 1])', 'blocks.2.0.conv_pwl.1.weight:torch.Size([32])', 'blocks.2.0.conv_pwl.1.bias:torch.Size([32])', 'blocks.2.0.conv_pwl.1.running_mean:torch.Size([32])', 'blocks.2.0.conv_pwl.1.running_var:torch.Size([32])', 'blocks.2.0.conv_pwl.1.num_batches_tracked:torch.Size([])', 'blocks.2.1.conv_pw.0.weight:torch.Size([192, 32, 1, 1])', 'blocks.2.1.conv_pw.1.weight:torch.Size([192])', 'blocks.2.1.conv_pw.1.bias:torch.Size([192])', 'blocks.2.1.conv_pw.1.running_mean:torch.Size([192])', 'blocks.2.1.conv_pw.1.running_var:torch.Size([192])', 'blocks.2.1.conv_pw.1.num_batches_tracked:torch.Size([])', 'blocks.2.1.conv_dw.0.weight:torch.Size([192, 1, 5, 5])', 'blocks.2.1.conv_dw.1.weight:torch.Size([192])', 'blocks.2.1.conv_dw.1.bias:torch.Size([192])', 'blocks.2.1.conv_dw.1.running_mean:torch.Size([192])', 'blocks.2.1.conv_dw.1.running_var:torch.Size([192])', 'blocks.2.1.conv_dw.1.num_batches_tracked:torch.Size([])', 'blocks.2.1.se.conv_reduce.weight:torch.Size([8, 192, 1, 1])', 'blocks.2.1.se.conv_reduce.bias:torch.Size([8])', 'blocks.2.1.se.conv_expand.weight:torch.Size([192, 8, 1, 1])', 'blocks.2.1.se.conv_expand.bias:torch.Size([192])', 'blocks.2.1.conv_pwl.0.weight:torch.Size([32, 192, 1, 1])', 'blocks.2.1.conv_pwl.1.weight:torch.Size([32])', 'blocks.2.1.conv_pwl.1.bias:torch.Size([32])', 'blocks.2.1.conv_pwl.1.running_mean:torch.Size([32])', 'blocks.2.1.conv_pwl.1.running_var:torch.Size([32])', 'blocks.2.1.conv_pwl.1.num_batches_tracked:torch.Size([])', 'blocks.3.0.conv_pw.0.weight:torch.Size([192, 32, 1, 1])', 'blocks.3.0.conv_pw.1.weight:torch.Size([192])', 'blocks.3.0.conv_pw.1.bias:torch.Size([192])', 'blocks.3.0.conv_pw.1.running_mean:torch.Size([192])', 'blocks.3.0.conv_pw.1.running_var:torch.Size([192])', 'blocks.3.0.conv_pw.1.num_batches_tracked:torch.Size([])', 'blocks.3.0.conv_dw.0.weight:torch.Size([192, 1, 3, 3])', 'blocks.3.0.conv_dw.1.weight:torch.Size([192])', 'blocks.3.0.conv_dw.1.bias:torch.Size([192])', 'blocks.3.0.conv_dw.1.running_mean:torch.Size([192])', 'blocks.3.0.conv_dw.1.running_var:torch.Size([192])', 'blocks.3.0.conv_dw.1.num_batches_tracked:torch.Size([])', 'blocks.3.0.se.conv_reduce.weight:torch.Size([8, 192, 1, 1])', 'blocks.3.0.se.conv_reduce.bias:torch.Size([8])', 'blocks.3.0.se.conv_expand.weight:torch.Size([192, 8, 1, 1])', 'blocks.3.0.se.conv_expand.bias:torch.Size([192])', 'blocks.3.0.conv_pwl.0.weight:torch.Size([64, 192, 1, 1])', 'blocks.3.0.conv_pwl.1.weight:torch.Size([64])', 'blocks.3.0.conv_pwl.1.bias:torch.Size([64])', 'blocks.3.0.conv_pwl.1.running_mean:torch.Size([64])', 'blocks.3.0.conv_pwl.1.running_var:torch.Size([64])', 'blocks.3.0.conv_pwl.1.num_batches_tracked:torch.Size([])', 'blocks.3.1.conv_pw.0.weight:torch.Size([384, 64, 1, 1])', 'blocks.3.1.conv_pw.1.weight:torch.Size([384])', 'blocks.3.1.conv_pw.1.bias:torch.Size([384])', 'blocks.3.1.conv_pw.1.running_mean:torch.Size([384])', 'blocks.3.1.conv_pw.1.running_var:torch.Size([384])', 'blocks.3.1.conv_pw.1.num_batches_tracked:torch.Size([])', 'blocks.3.1.conv_dw.0.weight:torch.Size([384, 1, 3, 3])', 'blocks.3.1.conv_dw.1.weight:torch.Size([384])', 'blocks.3.1.conv_dw.1.bias:torch.Size([384])', 'blocks.3.1.conv_dw.1.running_mean:torch.Size([384])', 'blocks.3.1.conv_dw.1.running_var:torch.Size([384])', 'blocks.3.1.conv_dw.1.num_batches_tracked:torch.Size([])', 'blocks.3.1.se.conv_reduce.weight:torch.Size([16, 384, 1, 1])', 'blocks.3.1.se.conv_reduce.bias:torch.Size([16])', 'blocks.3.1.se.conv_expand.weight:torch.Size([384, 16, 1, 1])', 'blocks.3.1.se.conv_expand.bias:torch.Size([384])', 'blocks.3.1.conv_pwl.0.weight:torch.Size([64, 384, 1, 1])', 'blocks.3.1.conv_pwl.1.weight:torch.Size([64])', 'blocks.3.1.conv_pwl.1.bias:torch.Size([64])', 'blocks.3.1.conv_pwl.1.running_mean:torch.Size([64])', 'blocks.3.1.conv_pwl.1.running_var:torch.Size([64])', 'blocks.3.1.conv_pwl.1.num_batches_tracked:torch.Size([])', 'blocks.3.2.conv_pw.0.weight:torch.Size([384, 64, 1, 1])', 'blocks.3.2.conv_pw.1.weight:torch.Size([384])', 'blocks.3.2.conv_pw.1.bias:torch.Size([384])', 'blocks.3.2.conv_pw.1.running_mean:torch.Size([384])', 'blocks.3.2.conv_pw.1.running_var:torch.Size([384])', 'blocks.3.2.conv_pw.1.num_batches_tracked:torch.Size([])', 'blocks.3.2.conv_dw.0.weight:torch.Size([384, 1, 3, 3])', 'blocks.3.2.conv_dw.1.weight:torch.Size([384])', 'blocks.3.2.conv_dw.1.bias:torch.Size([384])', 'blocks.3.2.conv_dw.1.running_mean:torch.Size([384])', 'blocks.3.2.conv_dw.1.running_var:torch.Size([384])', 'blocks.3.2.conv_dw.1.num_batches_tracked:torch.Size([])', 'blocks.3.2.se.conv_reduce.weight:torch.Size([16, 384, 1, 1])', 'blocks.3.2.se.conv_reduce.bias:torch.Size([16])', 'blocks.3.2.se.conv_expand.weight:torch.Size([384, 16, 1, 1])', 'blocks.3.2.se.conv_expand.bias:torch.Size([384])', 'blocks.3.2.conv_pwl.0.weight:torch.Size([64, 384, 1, 1])', 'blocks.3.2.conv_pwl.1.weight:torch.Size([64])', 'blocks.3.2.conv_pwl.1.bias:torch.Size([64])', 'blocks.3.2.conv_pwl.1.running_mean:torch.Size([64])', 'blocks.3.2.conv_pwl.1.running_var:torch.Size([64])', 'blocks.3.2.conv_pwl.1.num_batches_tracked:torch.Size([])', 'blocks.4.0.conv_pw.0.weight:torch.Size([384, 64, 1, 1])', 'blocks.4.0.conv_pw.1.weight:torch.Size([384])', 'blocks.4.0.conv_pw.1.bias:torch.Size([384])', 'blocks.4.0.conv_pw.1.running_mean:torch.Size([384])', 'blocks.4.0.conv_pw.1.running_var:torch.Size([384])', 'blocks.4.0.conv_pw.1.num_batches_tracked:torch.Size([])', 'blocks.4.0.conv_dw.0.weight:torch.Size([384, 1, 5, 5])', 'blocks.4.0.conv_dw.1.weight:torch.Size([384])', 'blocks.4.0.conv_dw.1.bias:torch.Size([384])', 'blocks.4.0.conv_dw.1.running_mean:torch.Size([384])', 'blocks.4.0.conv_dw.1.running_var:torch.Size([384])', 'blocks.4.0.conv_dw.1.num_batches_tracked:torch.Size([])', 'blocks.4.0.se.conv_reduce.weight:torch.Size([16, 384, 1, 1])', 'blocks.4.0.se.conv_reduce.bias:torch.Size([16])', 'blocks.4.0.se.conv_expand.weight:torch.Size([384, 16, 1, 1])', 'blocks.4.0.se.conv_expand.bias:torch.Size([384])', 'blocks.4.0.conv_pwl.0.weight:torch.Size([96, 384, 1, 1])', 'blocks.4.0.conv_pwl.1.weight:torch.Size([96])', 'blocks.4.0.conv_pwl.1.bias:torch.Size([96])', 'blocks.4.0.conv_pwl.1.running_mean:torch.Size([96])', 'blocks.4.0.conv_pwl.1.running_var:torch.Size([96])', 'blocks.4.0.conv_pwl.1.num_batches_tracked:torch.Size([])', 'blocks.4.1.conv_pw.0.weight:torch.Size([576, 96, 1, 1])', 'blocks.4.1.conv_pw.1.weight:torch.Size([576])', 'blocks.4.1.conv_pw.1.bias:torch.Size([576])', 'blocks.4.1.conv_pw.1.running_mean:torch.Size([576])', 'blocks.4.1.conv_pw.1.running_var:torch.Size([576])', 'blocks.4.1.conv_pw.1.num_batches_tracked:torch.Size([])', 'blocks.4.1.conv_dw.0.weight:torch.Size([576, 1, 5, 5])', 'blocks.4.1.conv_dw.1.weight:torch.Size([576])', 'blocks.4.1.conv_dw.1.bias:torch.Size([576])', 'blocks.4.1.conv_dw.1.running_mean:torch.Size([576])', 'blocks.4.1.conv_dw.1.running_var:torch.Size([576])', 'blocks.4.1.conv_dw.1.num_batches_tracked:torch.Size([])', 'blocks.4.1.se.conv_reduce.weight:torch.Size([24, 576, 1, 1])', 'blocks.4.1.se.conv_reduce.bias:torch.Size([24])', 'blocks.4.1.se.conv_expand.weight:torch.Size([576, 24, 1, 1])', 'blocks.4.1.se.conv_expand.bias:torch.Size([576])', 'blocks.4.1.conv_pwl.0.weight:torch.Size([96, 576, 1, 1])', 'blocks.4.1.conv_pwl.1.weight:torch.Size([96])', 'blocks.4.1.conv_pwl.1.bias:torch.Size([96])', 'blocks.4.1.conv_pwl.1.running_mean:torch.Size([96])', 'blocks.4.1.conv_pwl.1.running_var:torch.Size([96])', 'blocks.4.1.conv_pwl.1.num_batches_tracked:torch.Size([])', 'blocks.4.2.conv_pw.0.weight:torch.Size([576, 96, 1, 1])', 'blocks.4.2.conv_pw.1.weight:torch.Size([576])', 'blocks.4.2.conv_pw.1.bias:torch.Size([576])', 'blocks.4.2.conv_pw.1.running_mean:torch.Size([576])', 'blocks.4.2.conv_pw.1.running_var:torch.Size([576])', 'blocks.4.2.conv_pw.1.num_batches_tracked:torch.Size([])', 'blocks.4.2.conv_dw.0.weight:torch.Size([576, 1, 5, 5])', 'blocks.4.2.conv_dw.1.weight:torch.Size([576])', 'blocks.4.2.conv_dw.1.bias:torch.Size([576])', 'blocks.4.2.conv_dw.1.running_mean:torch.Size([576])', 'blocks.4.2.conv_dw.1.running_var:torch.Size([576])', 'blocks.4.2.conv_dw.1.num_batches_tracked:torch.Size([])', 'blocks.4.2.se.conv_reduce.weight:torch.Size([24, 576, 1, 1])', 'blocks.4.2.se.conv_reduce.bias:torch.Size([24])', 'blocks.4.2.se.conv_expand.weight:torch.Size([576, 24, 1, 1])', 'blocks.4.2.se.conv_expand.bias:torch.Size([576])', 'blocks.4.2.conv_pwl.0.weight:torch.Size([96, 576, 1, 1])', 'blocks.4.2.conv_pwl.1.weight:torch.Size([96])', 'blocks.4.2.conv_pwl.1.bias:torch.Size([96])', 'blocks.4.2.conv_pwl.1.running_mean:torch.Size([96])', 'blocks.4.2.conv_pwl.1.running_var:torch.Size([96])', 'blocks.4.2.conv_pwl.1.num_batches_tracked:torch.Size([])', 'blocks.5.0.conv_pw.0.weight:torch.Size([576, 96, 1, 1])', 'blocks.5.0.conv_pw.1.weight:torch.Size([576])', 'blocks.5.0.conv_pw.1.bias:torch.Size([576])', 'blocks.5.0.conv_pw.1.running_mean:torch.Size([576])', 'blocks.5.0.conv_pw.1.running_var:torch.Size([576])', 'blocks.5.0.conv_pw.1.num_batches_tracked:torch.Size([])', 'blocks.5.0.conv_dw.0.weight:torch.Size([576, 1, 5, 5])', 'blocks.5.0.conv_dw.1.weight:torch.Size([576])', 'blocks.5.0.conv_dw.1.bias:torch.Size([576])', 'blocks.5.0.conv_dw.1.running_mean:torch.Size([576])', 'blocks.5.0.conv_dw.1.running_var:torch.Size([576])', 'blocks.5.0.conv_dw.1.num_batches_tracked:torch.Size([])', 'blocks.5.0.se.conv_reduce.weight:torch.Size([24, 576, 1, 1])', 'blocks.5.0.se.conv_reduce.bias:torch.Size([24])', 'blocks.5.0.se.conv_expand.weight:torch.Size([576, 24, 1, 1])', 'blocks.5.0.se.conv_expand.bias:torch.Size([576])', 'blocks.5.0.conv_pwl.0.weight:torch.Size([160, 576, 1, 1])', 'blocks.5.0.conv_pwl.1.weight:torch.Size([160])', 'blocks.5.0.conv_pwl.1.bias:torch.Size([160])', 'blocks.5.0.conv_pwl.1.running_mean:torch.Size([160])', 'blocks.5.0.conv_pwl.1.running_var:torch.Size([160])', 'blocks.5.0.conv_pwl.1.num_batches_tracked:torch.Size([])', 'blocks.5.1.conv_pw.0.weight:torch.Size([960, 160, 1, 1])', 'blocks.5.1.conv_pw.1.weight:torch.Size([960])', 'blocks.5.1.conv_pw.1.bias:torch.Size([960])', 'blocks.5.1.conv_pw.1.running_mean:torch.Size([960])', 'blocks.5.1.conv_pw.1.running_var:torch.Size([960])', 'blocks.5.1.conv_pw.1.num_batches_tracked:torch.Size([])', 'blocks.5.1.conv_dw.0.weight:torch.Size([960, 1, 5, 5])', 'blocks.5.1.conv_dw.1.weight:torch.Size([960])', 'blocks.5.1.conv_dw.1.bias:torch.Size([960])', 'blocks.5.1.conv_dw.1.running_mean:torch.Size([960])', 'blocks.5.1.conv_dw.1.running_var:torch.Size([960])', 'blocks.5.1.conv_dw.1.num_batches_tracked:torch.Size([])', 'blocks.5.1.se.conv_reduce.weight:torch.Size([40, 960, 1, 1])', 'blocks.5.1.se.conv_reduce.bias:torch.Size([40])', 'blocks.5.1.se.conv_expand.weight:torch.Size([960, 40, 1, 1])', 'blocks.5.1.se.conv_expand.bias:torch.Size([960])', 'blocks.5.1.conv_pwl.0.weight:torch.Size([160, 960, 1, 1])', 'blocks.5.1.conv_pwl.1.weight:torch.Size([160])', 'blocks.5.1.conv_pwl.1.bias:torch.Size([160])', 'blocks.5.1.conv_pwl.1.running_mean:torch.Size([160])', 'blocks.5.1.conv_pwl.1.running_var:torch.Size([160])', 'blocks.5.1.conv_pwl.1.num_batches_tracked:torch.Size([])', 'blocks.5.2.conv_pw.0.weight:torch.Size([960, 160, 1, 1])', 'blocks.5.2.conv_pw.1.weight:torch.Size([960])', 'blocks.5.2.conv_pw.1.bias:torch.Size([960])', 'blocks.5.2.conv_pw.1.running_mean:torch.Size([960])', 'blocks.5.2.conv_pw.1.running_var:torch.Size([960])', 'blocks.5.2.conv_pw.1.num_batches_tracked:torch.Size([])', 'blocks.5.2.conv_dw.0.weight:torch.Size([960, 1, 5, 5])', 'blocks.5.2.conv_dw.1.weight:torch.Size([960])', 'blocks.5.2.conv_dw.1.bias:torch.Size([960])', 'blocks.5.2.conv_dw.1.running_mean:torch.Size([960])', 'blocks.5.2.conv_dw.1.running_var:torch.Size([960])', 'blocks.5.2.conv_dw.1.num_batches_tracked:torch.Size([])', 'blocks.5.2.se.conv_reduce.weight:torch.Size([40, 960, 1, 1])', 'blocks.5.2.se.conv_reduce.bias:torch.Size([40])', 'blocks.5.2.se.conv_expand.weight:torch.Size([960, 40, 1, 1])', 'blocks.5.2.se.conv_expand.bias:torch.Size([960])', 'blocks.5.2.conv_pwl.0.weight:torch.Size([160, 960, 1, 1])', 'blocks.5.2.conv_pwl.1.weight:torch.Size([160])', 'blocks.5.2.conv_pwl.1.bias:torch.Size([160])', 'blocks.5.2.conv_pwl.1.running_mean:torch.Size([160])', 'blocks.5.2.conv_pwl.1.running_var:torch.Size([160])', 'blocks.5.2.conv_pwl.1.num_batches_tracked:torch.Size([])', 'blocks.6.0.conv_pw.0.weight:torch.Size([960, 160, 1, 1])', 'blocks.6.0.conv_pw.1.weight:torch.Size([960])', 'blocks.6.0.conv_pw.1.bias:torch.Size([960])', 'blocks.6.0.conv_pw.1.running_mean:torch.Size([960])', 'blocks.6.0.conv_pw.1.running_var:torch.Size([960])', 'blocks.6.0.conv_pw.1.num_batches_tracked:torch.Size([])', 'blocks.6.0.conv_dw.0.weight:torch.Size([960, 1, 3, 3])', 'blocks.6.0.conv_dw.1.weight:torch.Size([960])', 'blocks.6.0.conv_dw.1.bias:torch.Size([960])', 'blocks.6.0.conv_dw.1.running_mean:torch.Size([960])', 'blocks.6.0.conv_dw.1.running_var:torch.Size([960])', 'blocks.6.0.conv_dw.1.num_batches_tracked:torch.Size([])', 'blocks.6.0.se.conv_reduce.weight:torch.Size([40, 960, 1, 1])', 'blocks.6.0.se.conv_reduce.bias:torch.Size([40])', 'blocks.6.0.se.conv_expand.weight:torch.Size([960, 40, 1, 1])', 'blocks.6.0.se.conv_expand.bias:torch.Size([960])', 'blocks.6.0.conv_pwl.0.weight:torch.Size([264, 960, 1, 1])', 'blocks.6.0.conv_pwl.1.weight:torch.Size([264])', 'blocks.6.0.conv_pwl.1.bias:torch.Size([264])', 'blocks.6.0.conv_pwl.1.running_mean:torch.Size([264])', 'blocks.6.0.conv_pwl.1.running_var:torch.Size([264])', 'blocks.6.0.conv_pwl.1.num_batches_tracked:torch.Size([])', 'conv_head.0.weight:torch.Size([1056, 264, 1, 1])', 'conv_head.1.weight:torch.Size([1056])', 'conv_head.1.bias:torch.Size([1056])', 'conv_head.1.running_mean:torch.Size([1056])', 'conv_head.1.running_var:torch.Size([1056])', 'conv_head.1.num_batches_tracked:torch.Size([])', 'classifier.weight:torch.Size([1000, 1056])', 'classifier.bias:torch.Size([1000])']