forward and outputs of models

[jimmykimmy68]

Hi, thanks for your great work!

I have two questions on the Bayesian CNN codes, which are as follows;

First, where is the forward method in the model class? e.g., in class BBBAlexNet, there only is init, but isn't 'forward' prediction method.

Second, in main_bayesian.py, how does the network (i.e., model) output both so-called "outputs" and "_kl" ?

[pluvias]

Hi. A little bit late, but still.

First, BBBAlexNet class is inherited from ModuleWrapper class that inherits from nn.Module and has a forward method. Second, this forward method returns 2 values.