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.