diff --git a/src/primaite/config/_package_data/training/training_config_main.yaml b/src/primaite/config/_package_data/training/training_config_main.yaml index 2cc29c55..0f99a501 100644 --- a/src/primaite/config/_package_data/training/training_config_main.yaml +++ b/src/primaite/config/_package_data/training/training_config_main.yaml @@ -5,7 +5,7 @@ # "SB3" (Stable Baselines3) # "RLLIB" (Ray RLlib) # "CUSTOM" (Custom Agent) -agent_framework: RLLIB +agent_framework: SB3 # Sets which deep learning framework will be used (by RLlib ONLY). # Default is TF (Tensorflow). @@ -35,20 +35,20 @@ hard_coded_agent_view: FULL # "NODE" # "ACL" # "ANY" node and acl actions -action_type: ACL +action_type: ANY # Number of episodes to run per session -num_episodes: 10 +num_episodes: 100 # Number of time_steps per episode num_steps: 256 # Sets how often the agent will save a checkpoint (every n time episodes). # Set to 0 if no checkpoints are required. Default is 10 -checkpoint_every_n_episodes: 5 +checkpoint_every_n_episodes: 100 # Time delay between steps (for generic agents) -time_delay: 10 +time_delay: 3 # Type of session to be run (TRAINING or EVALUATION) session_type: TRAINING