chunk_content: h.nn.modules.flatten.Flatten),
   ('head.drop', torch.nn.modules.linear.Identity),
   ('head.fc', torch.nn.modules.linear.Linear)]
   ]
  Upon closer inspection, we see that the 2D conv layers have names such as "stages.0.blocks.0.mlp.fc1" and
  "stages.0.blocks.0.mlp.fc2". How can we match those layer names specifically? You can write a regular
  expressions to match the layer names. For our case, the regex
  r".*\.mlp\.fc\d" should do the job.
  Furthermore,