generators module

Tools to build out-of-the-box trainers

create_dkl_trainer(d, model_type='realnvp', mask_type='checkerboard', model_params=None, device=device(type='cpu'), n_epochs=10, optim_class=None, lr=0.0001, **params)[source]

Create a dkl trainer with the a default flow model