def load(load_pars={}, **kw):
path = load_pars["outpath"]
model_pars = kw["model_pars"]
compute_pars = kw["compute_pars"]
data_pars = kw["data_pars"]
custom_pars = {
"MDN":
mdn.MDN,
"loss":
mdn.get_mixture_loss_func(model_pars["timesteps"],
model_pars["n_mixes"])
}
model0 = load_keras({"path": path + "/armdn.h5"}, custom_pars)
model = Model(model_pars=model_pars,
data_pars=data_pars,
compute_pars=compute_pars)
model.model = model0.model
session = None
return model, session
