oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

#1648 - Reverted benchmark sessions and episodes numbers

Browse Source
This commit is contained in:
Chris McCarthy
2023-07-21 15:07:21 +01:00
parent 63297ef0ed
commit a39541623a
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
benchmark/config/benchmark_training_config.yaml
View File

@@ -63,7 +63,7 @@ observation_space:
- name: LINK_TRAFFIC_LEVELS - name: LINK_TRAFFIC_LEVELS
# Number of episodes for training to run per session # Number of episodes for training to run per session
num_train_episodes: 10 num_train_episodes: 500
# Number of time_steps for training per episode # Number of time_steps for training per episode
num_train_steps: 256 num_train_steps: 256

2
benchmark/primaite_benchmark.py
View File

@@ -414,7 +414,7 @@ def run():
"""Run the PrimAITE benchmark.""" """Run the PrimAITE benchmark."""
start_datetime = datetime.now() start_datetime = datetime.now()
av_reward_per_episode_dicts = {} av_reward_per_episode_dicts = {}
for i in range(1, 3): for i in range(1, 11):
print(f"Starting Benchmark Session: {i}") print(f"Starting Benchmark Session: {i}")
with _get_benchmark_primaite_session() as session: with _get_benchmark_primaite_session() as session:
session.learn() session.learn()