Update agilerl version and simplified CNN definition

Farama-Foundation · Feb 9, 2024 · c767acf · c767acf
1 parent 6c8e8c1
commit c767acf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tutorials/AgileRL/agilerl_maddpg.py b/tutorials/AgileRL/agilerl_maddpg.py
@@ -23,8 +23,8 @@
     NET_CONFIG = {
         "arch": "cnn",  # Network architecture
         "h_size": [32, 32],  # Network hidden size
-        "c_size": [3, 32],  # CNN channel size
-        "k_size": [(1, 3, 3), (1, 3, 3)],  # CNN kernel size
+        "c_size": [32, 32],  # CNN channel size
+        "k_size": [3, 3],  # CNN kernel size
         "s_size": [2, 2],  # CNN stride size
         "normalize": True,  # Normalize image from range [0,255] to [0,1]
     }

diff --git a/tutorials/AgileRL/requirements.txt b/tutorials/AgileRL/requirements.txt
@@ -1,4 +1,4 @@
-agilerl>=0.1.16
+agilerl>=0.1.20
 pettingzoo[classic,atari,mpe]>=1.23.1
 SuperSuit>=3.9.0
 torch>=2.0.1