Loading and Evaluating a Trained Agent #12
ChrisMcCarthyDev
started this conversation in
Ideas
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
When a PrimAITE session is ran that outputs a trained PPO/A2C agent, the contents of the original training config file is copied and dumped into the session_metadata.json file. When an agent is loaded, PrimAITE extracts the training config and lay-down config from the session_metadata.json. Because of this, if the session type was a TRAIN_EVAL session (
session_type: TRAIN_EVAL
), the agent will go through a second round of further training before getting to the evaluation. This can be overridden by amending the session type in the metadata.json file.1. Train an Agent
This is done by calling
primaite session
:2. CD to the session output directory
3. Edit the session_metadata.json file
Edit the
session_type
key so that the value isEVAL
4. Run the PrimAITE Session and Load the Agent
Run `primaite session --load
Beta Was this translation helpful? Give feedback.
All reactions