Issue in running full sweep #2
Description
Is there any other configuration, we need to do before running the sweep for "RSGM ISM loss", "RSGM DSM loss". I get an error:
TypeError: Error instantiating 'riemannian_score_sde.sde.Brownian' : init() got an unexpected keyword argument 'beta_0'
Similar issue is observed when running High dimension torus example. Would it be possible to update the READme with more details?