@G.M
Is that because the code is not in-lined ? Would there be any easy way to un-roll the loop so that pytorch can better optimize it ?
The amount of networks I need is determined once and stays the same.