{ factor_string: -loss, # legacy code: original loss used in all previous experiments "sqrt_" + factor_string: -loss_sqrt, "unnorm_" + factor_string: -loss_unnorm, "unnorm_sqrt_" + factor_string: -loss_unnorm_sqrt } elif "r=" in factor_string: masks = datase...