Merge pull request #240 from NatalieHanzhina/main

Accelerator issue is fixed
This commit is contained in:
Shivam Shrirao 2023-06-08 13:24:20 +05:30 committed by GitHub
commit 2c29830075
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)