@albanD Unfortunately this isn’t an option for me anymore since I have a deadline pretty soon and repruning the networks takes several days. I might be able to change each layers parameters based on the size of the tensors within. But I still need a way to edit the original network’s dictionary in this case.