Method score

diff2flow/flow.py:236–239

(self, x, t, model, **model_kwargs)

        return model_output
    
    def score(self, x, t, model, **model_kwargs):
        # we only train velocity, hence only need to compute score from velocity
        score_out = self.schedule.get_score_from_velocity(model(x, t, **model_kwargs), x, t)
        return score_out

    def check_interval(self, diffusion_form="sigma", reverse=False, last_step_size=0.04):
        t0 = 0