diff --git a/tutorials/AgileRL/agilerl_maddpg.py b/tutorials/AgileRL/agilerl_maddpg.py index 902d6125f..af259381a 100644 --- 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 index 13459f060..7b1c60728 100644 --- 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