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

fix: exclude _is_pad mask from images

Browse Source
This commit is contained in:
Khalil Meftah
2026-07-24 09:28:04 +02:00
parent 39cdc4dea3
commit 0a342ad4d0
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/overfit_vf_variant.py
+1 -1
View File
@@ -152,7 +152,7 @@ def _collate_processed(samples):
} }
keys = { keys = {
*[key for key in samples[0] if key.startswith("observation.images.")], *[key for key in samples[0] if key.startswith("observation.images.") and not key.endswith("_is_pad")],
OBS_LANGUAGE_TOKENS, OBS_LANGUAGE_TOKENS,
OBS_LANGUAGE_ATTENTION_MASK, OBS_LANGUAGE_ATTENTION_MASK,
} }