fix readme (#273)

This commit is contained in:
蔡舒起 2021-01-16 19:27:35 +08:00 committed by GitHub
parent c6f2648e87
commit 866e35d550
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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"]}')
```