hmmm but will it be able to optimize trainable_net.W? Thats what Im mostly concerned. I will be as the output and then pass it to criterion and THEN backprop to optimize only thing in trainable_net.W. Will that work is my main concern…