admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-26 11:16:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove type exp

Browse Source
This commit is contained in:
Michel Aractingi
2026-01-14 17:09:56 +01:00
committed by Khalil Meftah
parent cb14264d2e
commit b1cfbac570
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/scripts/lerobot_train.py
+1 -1
View File
@@ -104,7 +104,7 @@ def update_policy(
if sample_weights is not None:
# Use per-sample loss for weighted training
# Note: Policies supporting sample weighting must implement forward(batch, reduction="none")
per_sample_loss, output_dict = policy.forward(batch, reduction="none") # type: ignore[call-arg]
per_sample_loss, output_dict = policy.forward(batch, reduction="none")
# Apply sample weights: L_weighted = Σ(w_i * l_i) / (Σw_i + ε)
# Weights are already normalized to sum to batch_size