Hi @lhkhiem28 , [["features.0", "features.1"]] should work.  More generally, you could print out the fully qualified name of every module in the model with something like
for name, mod in model.named_modules():
  print(name, mod)