Both reaching similar RMSEs as in MC Dropout paper.
Updated hidden_layers handling in all training files.