Hi,
I have a problem with reusing an nn.Module with a new architecture.
e.g.
class network(nn.Module)
def __init__(self, architecture):
     # architecture 1 details

def forward(self):
     # train network details


net = network(architecture)
# and optimizer and such

if __name__ == "__main__":
      
     # train architecture 1 
     output = net()

     architecture2 = create_new_arch()

     # create architecture 2
     del  net
     net = network(architecture2)

     # train architecture 2
     output = net()

Normally, this will train architecture 1 perfectly fine but when it comes to the second architecture, the error doesn’t seem to budge at all. I tried testing the architecture2 independently and it seems fine. How can I reuse the net object that I have with a new architecture?
Thank you.