Hi,
I want to change the forward behavior based on the epoch.  I can’t figure out how to pass either the epoch or a simple True/False to the network during training.
Should I make it a parameter?  If so can I change that in the training loop?
I did find a post that shows how to pass the value to forward, but if it’s a net module then I’m still defining the layers so I don’t see how to pass that value to the layers from the forward pass.
Thanks