-
Notifications
You must be signed in to change notification settings - Fork 1.2k
change API of train_fn and test_fn #229
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
Conversation
What is the rationale of changing the signature? Maybe BTW, I find |
d63a222
to
f280797
Compare
I agree with the idea of this PR. Just remember to change all occurrence. This PR changes the signature and thus breaks the compatibility, so it is easy to check whether you have changed all occurrence: just re-run all the examples. |
Yes, I've checked and all examples are runnable. |
train_fn(epoch) -> train_fn(epoch, num_env_step) test_fn(epoch) -> test_fn(epoch, num_env_step)
train_fn(epoch) -> train_fn(epoch, env_step)
test_fn(epoch) -> test_fn(epoch, env_step)