def nll_gen_model(self, X, stt): return diag_gauss_nll(X, stt, 1e-4)
def nll_recog_model(self, Z, stt): return diag_gauss_nll(Z, stt, 1e-4)