DHDRL
/

adaptive-wafer-rl

Reinforcement Learning

stable-baselines3

deep-reinforcement-learning

Model card Files Files and versions

DHDRL commited on Jul 23

Commit

d2a6431

·

verified ·

1 Parent(s): 11187cd

Update gru_belief_policy_v3_agnostic.py

Files changed (1) hide show

gru_belief_policy_v3_agnostic.py +1 -2

gru_belief_policy_v3_agnostic.py CHANGED Viewed

@@ -19,7 +19,7 @@ from gymnasium import spaces
 class BeliefGRUEncoder(nn.Module):
     """
     GRU belief module with resolution-agnostic convolutional spatial compression.
-    Uses adaptive pooling to handle ANY input resolution.
     """
     def __init__(
@@ -190,7 +190,6 @@ class BeliefGRUEncoder(nn.Module):
 class GRUAugmentedFeaturesExtractor(BaseFeaturesExtractor):
     """
     Resolution-agnostic convolutional spatial encoder + optional GRU belief module.
     Handles input images from 64×64 to 1024×1024+ without parameter explosion.
     Uses adaptive pooling to normalize spatial features to fixed size.
     """

 class BeliefGRUEncoder(nn.Module):
     """
     GRU belief module with resolution-agnostic convolutional spatial compression.
+    Uses adaptive pooling to handle any input resolution.
     """
     def __init__(
 class GRUAugmentedFeaturesExtractor(BaseFeaturesExtractor):
     """
     Resolution-agnostic convolutional spatial encoder + optional GRU belief module.
     Handles input images from 64×64 to 1024×1024+ without parameter explosion.
     Uses adaptive pooling to normalize spatial features to fixed size.
     """