Update train_dreambooth.py

This commit is contained in:
natkaha 2023-06-08 08:27:06 +04:00 committed by GitHub
parent 2933b6b20b
commit b320b59ff6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions

View File

@ -426,7 +426,7 @@ def main(args):
gradient_accumulation_steps=args.gradient_accumulation_steps,
mixed_precision=args.mixed_precision,
log_with="tensorboard",
logging_dir=logging_dir,
project_dir=logging_dir,
)
logging.basicConfig(
@ -866,4 +866,4 @@ def main(args):
if __name__ == "__main__":
args = parse_args()
main(args)
main(args)