Upload code/wavecurriculum_run/hf_push_login.py with huggingface_hub

code/wavecurriculum_run/hf_push_login.py

@@ -31,7 +31,10 @@ def _write_json(path, payload):
 _KEEP_HB = (
     "event", "step", "stage", "loss",
     "val_acc", "val_mass", "val_excess", "val_spread",
+    "val_mass_multi", "val_excess_multi", "val_out_multi", "val_kl_multi",
+    "val_hbound", "val_hgap", "val_hbound_puz",
     "loc_wave", "loc_coverage", "loc_lcs", "loc_dup",
+    "puzzle_acc", "last_ckpt_step",
 )