gym-examples 3.0.346__py3-none-any.whl → 3.0.348__py3-none-any.whl

gym_examples/__init__.py

@@ -5,4 +5,4 @@ register(
      entry_point="gym_examples.envs:WSNRoutingEnv",
 )
 
-__version__ = "3.0.346"
+__version__ = "3.0.348"

gym_examples/envs/wsn_env.py

@@ -39,7 +39,11 @@ class Attention(nn.Module):
 
     def forward(self, x):
         x = F.relu(self.linear1(x))  # Pass the input through a linear layer and a ReLU activation function
-        attention_weights = F.softmax(x, dim=0)  # Apply the softmax function to get the attention weights
+        print("==============================")
+        print(f"x.dim: {x.dim()}")
+        print("==============================")
+        raise ValueError("Stop here")
+        attention_weights = F.softmax(x, dim=1)  # Apply the softmax function to get the attention weights. Initially F.softmax(x, dim=0) 
         x = attention_weights * x  # Multiply the input by the attention weights
         x = self.linear2(x)  # Pass the result through another linear layer
         return x

{gym_examples-3.0.346.dist-info → gym_examples-3.0.348.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: gym-examples
-Version: 3.0.346
+Version: 3.0.348
 Summary: A custom environment for multi-agent reinforcement learning focused on WSN routing.
 Home-page: https://github.com/gedji/CODES.git
 Author: Georges Djimefo

gym_examples-3.0.348.dist-info/RECORD

@@ -0,0 +1,7 @@
+gym_examples/__init__.py,sha256=wF4URbgfzFU0k9BzXDG3rlWdNZK51fq3BjZVX24gw4g,166
+gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
+gym_examples/envs/wsn_env.py,sha256=K6GSlHNrO4JGYQXXX5VH2Avs2deW2RHc54UTkOWWR4Y,25359
+gym_examples-3.0.348.dist-info/METADATA,sha256=r5kFwvHGCVq_LU7ML-QNM5AuHuJdLxLEI83Bv3IA6Xk,412
+gym_examples-3.0.348.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
+gym_examples-3.0.348.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
+gym_examples-3.0.348.dist-info/RECORD,,

gym_examples-3.0.346.dist-info/RECORD

@@ -1,7 +0,0 @@
-gym_examples/__init__.py,sha256=xOg111Pr4KltxWo_ztJ6rZGocIZRaABDoDNdXXRwC9E,166
-gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
-gym_examples/envs/wsn_env.py,sha256=eUT4sHZe51N_G_fgw1xfgzV34dbelhoHraGC1sHcM_M,25154
-gym_examples-3.0.346.dist-info/METADATA,sha256=ri0W7535fc5hVAEblnI70WNHh_lOmLt1GAp2-E-t8PQ,412
-gym_examples-3.0.346.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-gym_examples-3.0.346.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
-gym_examples-3.0.346.dist-info/RECORD,,