Skip to content

Conversation

@JunnYu
Copy link
Member

@JunnYu JunnYu commented Nov 15, 2022

PR types

Others

PR changes

Others

Description

  • 更新ldm train代码。
  • 支持ema。
  • 支持根据config初始化模型权重。

@JunnYu JunnYu requested a review from ZHUI November 16, 2022 08:07
@JunnYu JunnYu requested review from guoshengCS and removed request for ZHUI November 16, 2022 08:26
@JunnYu JunnYu requested a review from ZHUI November 16, 2022 08:52
@JunnYu JunnYu requested a review from ZHUI November 16, 2022 10:30
firestonelib
firestonelib previously approved these changes Nov 17, 2022
Copy link
Contributor

@firestonelib firestonelib left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@JunnYu JunnYu merged commit 0ce21ea into PaddlePaddle:develop Nov 17, 2022
if self.args.should_save:
self._rotate_checkpoints(use_mtime=True, output_dir=run_dir)

def set_optimizer_grouped_parameters(self,
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

加注释

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants