Skip to content

GitLab

Unverified Commit c7bfb8b2 authored by Wang, Yi's avatar Wang, Yi Committed by GitHub
Browse files

set the model to train state before accelerator prepare (#5099) 


Signed-off-by: default avatarWang, Yi A <yi.a.wang@intel.com>
parent 67d07074
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
examples/textual_inversion/textual_inversion.py
View file @ c7bfb8b2
...@@ -761,6 +761,7 @@ def main(): ...@@ -761,6 +761,7 @@ def main():
num_cycles=args.lr_num_cycles, num_cycles=args.lr_num_cycles,
) )
text_encoder.train()
# Prepare everything with our `accelerator`. # Prepare everything with our `accelerator`.
text_encoder, optimizer, train_dataloader, lr_scheduler = accelerator.prepare( text_encoder, optimizer, train_dataloader, lr_scheduler = accelerator.prepare(
text_encoder, optimizer, train_dataloader, lr_scheduler text_encoder, optimizer, train_dataloader, lr_scheduler
......
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