Skip to content

relax learner state to be a general dictionary#8

Merged
zhiyun merged 3 commits intoapple:mainfrom
zhiyun:learner_state
Jul 23, 2023
Merged

relax learner state to be a general dictionary#8
zhiyun merged 3 commits intoapple:mainfrom
zhiyun:learner_state

Conversation

@zhiyun
Copy link
Copy Markdown
Contributor

@zhiyun zhiyun commented Jul 21, 2023

Relax learner state to be a dictionary that can contain arbitrary keys.

Remove ema in learner state when ema.decay is None.

@zhiyun zhiyun requested review from markblee and ruomingp July 22, 2023 18:24
@zhiyun zhiyun merged commit b96e3ff into apple:main Jul 23, 2023
@zhiyun zhiyun deleted the learner_state branch July 23, 2023 02:15
apghml pushed a commit to apghml/axlearn that referenced this pull request Oct 3, 2023
5cp pushed a commit to 5cp/axlearn that referenced this pull request May 11, 2024
Add scripts/yaml/readme for running on EKS
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.

2 participants