VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: wandb not working without logging_dir

Browse Source
This commit is contained in:
Plat
2023-04-20 05:14:28 +09:00
parent 27ffd9fe3d
commit 8090daca40
1 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
library/train_util.py
View File

@@ -2740,17 +2740,20 @@ def load_tokenizer(args: argparse.Namespace):
def prepare_accelerator(args: argparse.Namespace): def prepare_accelerator(args: argparse.Namespace):
if args.logging_dir is None: if args.logging_dir is None:
log_with = None
logging_dir = None logging_dir = None
else: else:
log_prefix = "" if args.log_prefix is None else args.log_prefix
logging_dir = args.logging_dir + "/" + log_prefix + time.strftime("%Y%m%d%H%M%S", time.localtime())
log_with = "tensorboard" if args.log_with is None else args.log_with log_with = "tensorboard" if args.log_with is None else args.log_with
if log_with in ["wandb", "all"]: if log_with in ["wandb", "all"]:
if logging_dir is not None:
os.makedirs(logging_dir)
os.environ["WANDB_DIR"] = logging_dir
try: try:
import wandb import wandb
except ImportError: except ImportError:
raise ImportError("No wandb / wandb がインストールされていないようです") raise ImportError("No wandb / wandb がインストールされていないようです")
log_prefix = "" if args.log_prefix is None else args.log_prefix
logging_dir = args.logging_dir + "/" + log_prefix + time.strftime("%Y%m%d%H%M%S", time.localtime())
accelerator = Accelerator( accelerator = Accelerator(
gradient_accumulation_steps=args.gradient_accumulation_steps, gradient_accumulation_steps=args.gradient_accumulation_steps,