Brian Egolf

> > What should one do, if one wants to get both the mean and the stddev ? Thanks in advance. > > TF Version 2.13.0 TFP Version 0.20.1 >...

This solved it for me ```python def NLL(y, distr): return -distr.log_prob(y) def my_dist(params): return tfd.Normal(loc=params[:,0:1], scale=1e-3 + tf.math.softplus(0.05 * params[:,1:2]))# both parameters are learnable inputs = Input(shape=(1,)) out1 = Dense(1)(inputs)...

same issue on jupyter 3.6.6. installing from Git did not solve the issue

> I have the same problem on 5700XT using rocm 5.4.2 and pytorch 2.0 Strangely, it works fine using pytorch 1.13.1 > > same issue with both --medvram and --lowvram...