Skip to content
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

[Feature request] Hyperparameter optimization from pretrained agent #91

Open
Jonathan2021 opened this issue May 11, 2021 · 2 comments
Open
Labels
enhancement New feature or request

Comments

@Jonathan2021
Copy link

Enabling the possibility to run --optimize with the --trained-agent flag would be great !
In my case, I pre-trained an agent on a simplified task and want to continue training it on the real task (which involves a modified reward, more obstacles etc.).
It would be great to be able to run a hyperparameter search for this second phase of the training. (Even though some hyperparameters, such as the network architecture, can't be tuned here).
For now, when I run both flags together, it just continues training (weirdly outputting less info than without the --optimize flag by the way).
Thanks for the awesome training framework !

@araffin araffin added the enhancement New feature or request label May 11, 2021
@araffin
Copy link
Member

araffin commented May 12, 2021

Hello,
I'm unsure about such feature.
On one side, it seems to be a reasonable (even though unconventional) request. On the other side there are some behaviors that may be ill-defined.
For instance, if your pre-trained agent has a replay buffer size of 1e6, you should not change that during hyperparameter optimization. The same goes with other hyperparameters.

As a compromise you can fork this repo and create the feature in it (and post a link there ;)).

@Jonathan2021
Copy link
Author

Not sure I will have enough time nor the courage to do so but maybe. Thanks for the reply anyways ;)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

2 participants