Hi Peter,
Thanks for getting back to me! I’ve tried using self.named_modules() but I’m not sure how to specify the name.
When I say for m in self.named_modules() and print m’s in that loop, I get this

layers2.PNG783×491 138 KB

I’ve tried saying if m=="fc1a" or if m==fc1a and a bunch of other things in the named_modules loop but it never works. I’ve looked at the documentation page but it didn’t help. I’ve also tried to refer to the sequential fragment but, once again, I don’t really know how to refer to “encoder.0” etc
How exactly do I specify the names in the named_modules loop?