Zan Klanecek

Input argument for ELBO criterion

3

https://github.com/kumar-shridhar/PyTorch-BayesianCNN/blob/d93bad543c3226cd0fe05c0cb0ba033c41b3caa6/main_bayesian.py#L116 Your training set is divided on train and validation. Shouldn't here be `(1-valid_size)*len(trainset)`?