Skip to content

GitLab

"configs/git@developer.sourcefind.cn:OpenDAS/opencompass.git" did not exist on "89a8a8917b3135dec68c8de0922cdcb6e0f557a2"
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
Show 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