Hi All,
I just have a quick question regarding RecursiveScriptModule, I’ve built a custom optimizer and within it I cycle through all layers via for module in net.modules() and call the name of the module via module.__class__.__name___. I need this as my optimizer handles different layers differently, and I determine what type of layer it is via module.__class__.__name___.
However, if I jit my network this fails because all the name of all the modules become RecursiveScriptModule, I did notice that when printing out module it has RecursiveScriptModule(original_name=Linear). So, I was wondering how can I get the orginal_name variable here?
Thanks in advance!