-
Notifications
You must be signed in to change notification settings - Fork 373
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
feature(zjow): add qgpo policy for new DI-engine pipeline #757
Conversation
Codecov ReportAttention:
Additional details and impacted files@@ Coverage Diff @@
## main #757 +/- ##
==========================================
- Coverage 76.82% 75.80% -1.03%
==========================================
Files 676 679 +3
Lines 54328 55272 +944
==========================================
+ Hits 41737 41897 +160
- Misses 12591 13375 +784
Flags with carried forward coverage won't be shown. Click here to find out more. ☔ View full report in Codecov by Sentry. |
ding/policy/qgpo.py
Outdated
self.qt_update_momentum * param.data + (1 - self.qt_update_momentum) * target_param.data | ||
) | ||
|
||
q0_loss = q0_loss.detach().cpu().numpy() |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
replace detach() by item()
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
fixed.
@@ -611,6 +611,9 @@ def _plot(ctx: "OfflineRLContext"): | |||
) | |||
|
|||
if ctx.eval_value != -np.inf: | |||
if hasattr(ctx, "info_for_logging"): | |||
info_for_logging.update(ctx.info_for_logging) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
add comment for it
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
fixed.
Add qgpo policy for new DI-engine pipeline based on the implementation of repo https://github.com/ChenDRAG/CEP-energy-guided-diffusion.