From c767acff6d2313249e92f488c1b23ef076a65bbf Mon Sep 17 00:00:00 2001 From: nicku-a Date: Fri, 9 Feb 2024 10:57:25 +0000 Subject: [PATCH] Update agilerl version and simplified CNN definition --- tutorials/AgileRL/agilerl_maddpg.py | 4 ++-- tutorials/AgileRL/requirements.txt | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) 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