Skip to content

GitLab

Commit e9eb0938 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

make style

parent a29ea36d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/unconditional_image_generation/train_unconditional.py
View file @ e9eb0938
......@@ -287,7 +287,7 @@ def main(args):
logging_dir = os.path.join(args.output_dir, args.logging_dir)
accelerator_project_config = ProjectConfiguration(project_dir=args.output_dir, logging_dir=logging_dir)
kwargs = InitProcessGroupKwargs(timeout=timedelta(seconds=7200))#a big number for high resolution or big dataset
kwargs = InitProcessGroupKwargs(timeout=timedelta(seconds=7200)) # a big number for high resolution or big dataset
accelerator = Accelerator(
gradient_accumulation_steps=args.gradient_accumulation_steps,
mixed_precision=args.mixed_precision,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment