alibaba / alibaba/Tora

forward_step_eval Error

Open
#37 1 comment 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
1.2k
Forks
61
PR merge metrics
No merged PRs in 30d

Description

I encountered an error while executing eval during the model training process.
[rank2]: objects_to_broadcast = [batch["txt"], broadcast_shape, batch["points"]] # 准备广播对象
[rank2]: KeyError: 'points'`

But I checked the data loaded in each batch, and it only contains 4 keys. This is defined in the **SFTDataset**:
`item = {
"mp4": tensor_frms,
"txt": caption,
"num_frames": num_frames,
"fps": fps,
}`

Contributor guide

No contributing guide indexed for this repository

Research direction

Start at the forward_step_eval entry point and compare its expected batch keys with the SFTDataset item definition shown in the issue. Trace where the batch is assembled and determine which interface is intended; done means evaluation no longer raises KeyError for points and the dataset/evaluation contract is consistent.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Bug
Difficulty
3/5
Estimated time
1-2 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.