Merge pull request #233 from ftgreat/gpm_dev

disable wandb
FlagAI-Open · Feb 27, 2023 · acbb969 · acbb969
2 parents f589a8f + 40f65f2
commit acbb969
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/flagai/env_trainer_v1.py b/flagai/env_trainer_v1.py
@@ -120,6 +120,7 @@ def __init__(self,
 
         # wandb
         self.wandb = env_args.wandb
+        self.wandb = False
 
         # if model already_fp16, OPT 1.3B
         self.already_fp16 = env_args.already_fp16

diff --git a/flagai/trainer_v1.py b/flagai/trainer_v1.py
@@ -225,7 +225,8 @@ def __init__(
         self.training_script = training_script
 
         # wandb
-        self.wandb = wandb
+        self.wandb = env_args.wandb
+        self.wandb = False
 
         # if model already_fp16, OPT 1.3B
         self.already_fp16 = already_fp16