Eval and train

how to set train model and evaluation mode in train, normally in pytorch, we set model.train(), model.eval() to automatically stop all dropout, how to stop dropout in pyro in evaluation

not sure what you’re asking. train and eval in pytorch are methods for nn.Modules. if you use them in your pyro model, then you would set them the same way.

1 Like