The model looks fine to me - maybe try changing the TruncatedNormal to HalfNormal?
TruncatedNormal
HalfNormal