@LeviViana  The problem is I currently have about 20 weight files of pruned networks (more on the way) so recreating every network by hand isn’t really doable. The goal is to read a network from a config file, edit the tensor dimensions to be the same as the pruned weight file and then load in the weights.