From 40f65f29757e693c23ab804afd23250ff80f1ce8 Mon Sep 17 00:00:00 2001 From: ftgreat Date: Mon, 27 Feb 2023 09:21:23 +0800 Subject: [PATCH] disable wandb Signed-off-by: ftgreat --- flagai/env_trainer_v1.py | 1 + flagai/trainer_v1.py | 3 ++- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/flagai/env_trainer_v1.py b/flagai/env_trainer_v1.py index ffc6d8bf..a079674c 100644 --- 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 index e3615005..67bacb6f 100644 --- 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