Upload modeling_prismatic.py
Browse files- modeling_prismatic.py +2 -0
modeling_prismatic.py
CHANGED
|
@@ -559,4 +559,6 @@ class OpenVLAForActionPrediction(PrismaticForConditionalGeneration):
|
|
| 559 |
def get_action_stats(self, unnorm_key: Optional[str] = None) -> Dict[str, Any]:
|
| 560 |
"""Get all the logged statistics for the given dataset."""
|
| 561 |
unnorm_key = self._check_unnorm_key(self.norm_stats, unnorm_key)
|
|
|
|
|
|
|
| 562 |
return self.norm_stats[unnorm_key]["action"]
|
|
|
|
| 559 |
def get_action_stats(self, unnorm_key: Optional[str] = None) -> Dict[str, Any]:
|
| 560 |
"""Get all the logged statistics for the given dataset."""
|
| 561 |
unnorm_key = self._check_unnorm_key(self.norm_stats, unnorm_key)
|
| 562 |
+
print("keys:", self.norm_stats.keys())
|
| 563 |
+
print("items:", self.norm_stats)
|
| 564 |
return self.norm_stats[unnorm_key]["action"]
|