wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix typo

This commit is contained in:
jayLEE0301 2024-05-24 20:04:14 -04:00
parent 547d3c3ea1
commit 5f6b66562f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/common/policies/vqbet/modeling_vqbet.py
View File

@ -801,7 +801,7 @@ class VqVae(nn.Module):
dec_out = self.decoder(state_vq) dec_out = self.decoder(state_vq)
encoder_loss = (state - dec_out).abs().mean() encoder_loss = (state - dec_out).abs().mean()
rep_loss = encoder_loss * vq_loss_state * 5 rep_loss = encoder_loss + vq_loss_state * 5
metric = ( metric = (
encoder_loss.clone().detach(), encoder_loss.clone().detach(),