bobox's picture
n_layers_per_step = 1, last_layer_weight = 1 * model_layers,, prior_layers_weight= 0.05, kl_div_weight = 2, kl_temperature= 0.9,
aa1484c verified