Add option for fixed (not learnable) random field parameters in BatchGaussianRandomFieldModule
. These parameters will often be (effectively) constant in space/time, which means there are initialized with huge (correlation) values. As such, keeping things stable also required using jnp.expm1
to implement (1 - self.phi**2)
in the GaussianRandomField
.
#292
Job | Run time |
---|---|
5m 40s | |
6m 32s | |
12m 12s |