(num_steps, metrics)
| 180 | |
| 181 | |
| 182 | def progress(num_steps, metrics): |
| 183 | times.append(datetime.now()) |
| 184 | xdata.append(num_steps) |
| 185 | ydata.append(metrics["eval/episode_reward"]) |
| 186 | print(f"step: {num_steps}, episode return: {metrics['eval/episode_reward']:.2f}") |
| 187 | |
| 188 | |
| 189 | make_inference_fn, params, _ = train_fn(environment=env, progress_fn=progress) |
nothing calls this directly
no outgoing calls
no test coverage detected