def METHOD_NAME(lbann):
    latent_dim = 2048
    number_of_atoms = 11
    layers, img_loss, metrics = MOFae.gen_layers(latent_dim, number_of_atoms)
    callbacks = [lbann.CallbackPrint(), lbann.CallbackTimer()]
    return lbann.Model(num_epochs,
                       layers = layers, 
                       objective_function = img_loss, 
                       metrics = metrics, 
                       callbacks = callbacks
                       )