{"trained_orig": {"kl_div": 0.0, "nll": 1.0137733221054077, "match_nll": 0.0}, "trained_patched": {"kl_div": 15.689358711242676, "nll": 16.729799270629883, "match_nll": 15.716024398803711}, "reset_orig": {"kl_div": 15.230734825134277, "nll": 16.419483184814453, "match_nll": 15.405710220336914}, "reset_patched": {"kl_div": 16.83327865600586, "nll": 17.941612243652344, "match_nll": 16.927837371826172}}