Mava icon indicating copy to clipboard operation
Mava copied to clipboard

Published 20 hours ago • verified publisher icon instadeepai

Use `TrainState` in evaluator

Open sash-a opened this issue 5 months ago • 0 comments

As discussed in #994 we should now be able to use the MetricsWrapper and the TrainState in the evaluator instead of manually recording the episode return and creating a special EvalState.

sash-a avatar Jan 26 '24 07:01 sash-a