From 866e35d550fe9aaaff5492f463eb59d9196efc83 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=94=A1=E8=88=92=E8=B5=B7?= <71307974+StevenJokess@users.noreply.github.com> Date: Sat, 16 Jan 2021 19:27:35 +0800 Subject: [PATCH] fix readme (#273) --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 833a856..36a379b 100644 --- a/README.md +++ b/README.md @@ -234,7 +234,7 @@ result = ts.trainer.offpolicy_trainer( train_fn=lambda epoch, env_step: policy.set_eps(eps_train), test_fn=lambda epoch, env_step: policy.set_eps(eps_test), stop_fn=lambda mean_rewards: mean_rewards >= env.spec.reward_threshold, - writer=writer, task=task) + writer=writer) print(f'Finished training! Use {result["duration"]}') ```