From 5f6b66562f5e0766658181c10402c015c3f16e61 Mon Sep 17 00:00:00 2001 From: jayLEE0301 Date: Fri, 24 May 2024 20:04:14 -0400 Subject: [PATCH] fix typo --- lerobot/common/policies/vqbet/modeling_vqbet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lerobot/common/policies/vqbet/modeling_vqbet.py b/lerobot/common/policies/vqbet/modeling_vqbet.py index c5ab1797..7271150c 100644 --- a/lerobot/common/policies/vqbet/modeling_vqbet.py +++ b/lerobot/common/policies/vqbet/modeling_vqbet.py @@ -801,7 +801,7 @@ class VqVae(nn.Module): dec_out = self.decoder(state_vq) encoder_loss = (state - dec_out).abs().mean() - rep_loss = encoder_loss * vq_loss_state * 5 + rep_loss = encoder_loss + vq_loss_state * 5 metric = ( encoder_loss.clone().detach(),