Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
renzhc
diffusers_dcu
Commits
37b359b2
Unverified
Commit
37b359b2
authored
Apr 06, 2023
by
Kadir Nar
Committed by
GitHub
Apr 06, 2023
Browse files
The variable name has been updated. (#2970)
parent
a9477bbd
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint.py
...h_projects/dreambooth_inpaint/train_dreambooth_inpaint.py
+2
-2
No files found.
examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint.py
View file @
37b359b2
...
...
@@ -405,14 +405,14 @@ def main():
args
=
parse_args
()
logging_dir
=
Path
(
args
.
output_dir
,
args
.
logging_dir
)
accelerator_
project_config
=
ProjectConfiguration
(
total_limit
=
args
.
checkpoints_total_limit
)
project_config
=
ProjectConfiguration
(
total_limit
=
args
.
checkpoints_total_limit
)
accelerator
=
Accelerator
(
gradient_accumulation_steps
=
args
.
gradient_accumulation_steps
,
mixed_precision
=
args
.
mixed_precision
,
log_with
=
"tensorboard"
,
logging_dir
=
logging_dir
,
accelerator_
project_config
=
accelerator_
project_config
,
project_config
=
project_config
,
)
# Currently, it's not possible to do gradient accumulation when training two models with accelerate.accumulate
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment