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

Add args to configure the training runtime + save training information and results. #8

Open
5 of 8 tasks
hidal00p opened this issue Jul 3, 2024 · 1 comment
Open
5 of 8 tasks
Assignees

Comments

@hidal00p
Copy link
Owner

hidal00p commented Jul 3, 2024

Args

  • Expose argparse configuration.
  • Add algorithm selection.
  • ~number of training episodes.
  • max episode length.
  • n cpus

Save format

  • Metadata should be human readable (a pretty compact .txt information).
  • Model is saved in a binary format as per stable baselines implementation.
  • This should enable a loadable scenario.
@hidal00p hidal00p self-assigned this Jul 3, 2024
@hidal00p hidal00p moved this to In Progress in i-hate-drones Jul 4, 2024
@hidal00p hidal00p changed the title Specify saving format for trained models. Add args to configure the training runtime + save training information and results. Jul 4, 2024
@hidal00p
Copy link
Owner Author

hidal00p commented Jul 5, 2024

This would probably be my final task apart from potential reward function tweaking.
I would like some easy API to load the whole env i.e.

  • Trajectory
  • Obstacle locations
  • Model
  • Eyes

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
Status: In Progress
Development

No branches or pull requests

1 participant