Merge pull request #240 from NatalieHanzhina/main
Accelerator issue is fixed
This commit is contained in:
commit
2c29830075
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue