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.__class__.__name___. I need this as my optimizer handles different layers differently, and I determine what type of layer it is via
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!