mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-23 17:56:07 +00:00
iterate on cpilot
This commit is contained in:
@@ -280,7 +280,6 @@ class FrankaJoint7ActionSpace(BaseActionSpace):
|
|||||||
|
|
||||||
def compute_loss(self, pred, target):
|
def compute_loss(self, pred, target):
|
||||||
assert pred.shape == target.shape, "pred/target shapes must match"
|
assert pred.shape == target.shape, "pred/target shapes must match"
|
||||||
batch_size, seq_len, action_dim = pred.shape
|
|
||||||
joints_loss = self.mse(pred, target) * self.JOINTS_SCALE
|
joints_loss = self.mse(pred, target) * self.JOINTS_SCALE
|
||||||
return {"joints_loss": joints_loss}
|
return {"joints_loss": joints_loss}
|
||||||
|
|
||||||
|
|||||||
@@ -345,7 +345,6 @@ class Florence2Config(PretrainedConfig):
|
|||||||
if vision_config is not None:
|
if vision_config is not None:
|
||||||
vision_config = Florence2VisionConfig(**vision_config)
|
vision_config = Florence2VisionConfig(**vision_config)
|
||||||
self.vision_config = vision_config
|
self.vision_config = vision_config
|
||||||
self.vocab_size = self.vocab_size
|
|
||||||
|
|
||||||
self.text_config = text_config
|
self.text_config = text_config
|
||||||
if text_config is not None:
|
if text_config is not None:
|
||||||
|
|||||||
Reference in New Issue
Block a user