Skip to content

Support actor model averaging for ActorCriticAlgorithm#1820

Merged
emailweixu merged 2 commits intopytorchfrom
PR_ac_ema
Nov 12, 2025
Merged

Support actor model averaging for ActorCriticAlgorithm#1820
emailweixu merged 2 commits intopytorchfrom
PR_ac_ema

Conversation

@emailweixu
Copy link
Contributor

model averaging is often helpful for evaluation.

model averaging is often helpful for evaluation.
loss_class (type): the class of the loss. The signature of its
constructor: ``loss_class(debug_summaries)``
actor_avg_fns: a list of functions for doing model averaging for
the trajectory predictor. Each function will be responsible for
Copy link
Contributor

@Haichao-Zhang Haichao-Zhang Oct 25, 2025

Choose a reason for hiding this comment

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

trajectory predictor: do you mean the actor in general?

@emailweixu emailweixu merged commit 160c6fa into pytorch Nov 12, 2025
2 checks passed
@emailweixu emailweixu deleted the PR_ac_ema branch November 12, 2025 19:49
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