Hi @ptrblck, thanks for your suggestion and apologies for delayed reply. I instead created the network class that will take the network hyperparameters like num of output channels, filter size etc. as input param hence I need not reset the network each time