login to wandb

ku-nlp · Dec 14, 2023 · 921989e · 921989e
1 parent d22aab6
commit 921989e
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/llm_judge/gen_judgment.py b/llm_judge/gen_judgment.py
@@ -145,6 +145,11 @@ def make_match_pairwise(
     parser.add_argument(
         "--overwrite", action="store_true", help="Overwrite existing judgment files."
     )
+    parser.add_argument(
+        "--wandb",
+        action="store_true",
+        help="Log to wandb.",
+    )
     args = parser.parse_args()
 
     if args.verbose == 0:
@@ -157,6 +162,12 @@ def make_match_pairwise(
         level=level,
     )
 
+    if args.wandb:
+        import wandb
+
+        wandb.login()
+        wandb.init()
+
     logger.info("Load questions")
     questions = load_questions(str(QUESTION_FILE))
     if args.first_n: