Skip to content

[kandinsky][train_text_to_image_prior.py] Fix LR scheduler when num_train_epochs is passed in a distributed training env - #14597

Open
adi-IL wants to merge 1 commit into
huggingface:mainfrom
adi-IL:fix/kandinsky-prior-lr-scheduler
Open

[kandinsky][train_text_to_image_prior.py] Fix LR scheduler when num_train_epochs is passed in a distributed training env#14597
adi-IL wants to merge 1 commit into
huggingface:mainfrom
adi-IL:fix/kandinsky-prior-lr-scheduler

fix(kandinsky): correct LR scheduler steps for prior in distributed t…

63e4801
Select commit
Loading
Failed to load commit list.

Select a check to view from the sidebar