Hey @anqipan,
I am finding it hard to understand your post, would it be possible for you to share the data so that I can run the model locally and get back to you?
Otherwise could you share the complete stack trace for “1”? The errors in the first post and the second look different, so I am not sure where the problem lies.
For “3” you don’t need to pass in the dimensions to PDiagDomain it should just be Ω ∈ PDiagDomain(init=[0.25,0.25,0.25])