Update modeling_prismatic.py
Browse files- modeling_prismatic.py +1 -0
modeling_prismatic.py
CHANGED
|
@@ -553,6 +553,7 @@ class OpenVLAForActionPrediction(PrismaticForConditionalGeneration):
|
|
| 553 |
def get_action_dim(self, unnorm_key=None):
|
| 554 |
"""Dimensionality of the policy's action space."""
|
| 555 |
unnorm_key = self._check_unnorm_key(self.norm_stats, unnorm_key)
|
|
|
|
| 556 |
return len(self.norm_stats[unnorm_key]["action"]["q01"])
|
| 557 |
|
| 558 |
def get_action_stats(self, unnorm_key=None):
|
|
|
|
| 553 |
def get_action_dim(self, unnorm_key=None):
|
| 554 |
"""Dimensionality of the policy's action space."""
|
| 555 |
unnorm_key = self._check_unnorm_key(self.norm_stats, unnorm_key)
|
| 556 |
+
|
| 557 |
return len(self.norm_stats[unnorm_key]["action"]["q01"])
|
| 558 |
|
| 559 |
def get_action_stats(self, unnorm_key=None):
|