Sitheral

I've figured out a fix for this. You might try modifying the model and guide function in the following manner def model(x_data, y_data): fc1w_prior = Normal(loc=torch.zeros_like(net.fc1.weight), scale=torch.ones_like(net.fc1.weight)).to_event(2) fc1b_prior = Normal(loc=torch.zeros_like(net.fc1.bias),...