Skip to content

Commit

Permalink
set dummy data fix length false in ci (#163)
Browse files Browse the repository at this point in the history
  • Loading branch information
sunpengsdu authored Apr 1, 2024
1 parent daa7338 commit 892862e
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions tests/test_training/test_loss.py
Original file line number Diff line number Diff line change
Expand Up @@ -91,6 +91,7 @@ def train(

# init setting
gpc.config.data.total_steps = TOTAL_STEPS
gpc.config.data.fixed_random_dataset_seqlen = False
gpc.config.lr_scheduler.total_steps = TOTAL_STEPS
gpc.config.model_type = model_type
total_steps = gpc.config.data.total_steps
Expand Down

0 comments on commit 892862e

Please sign in to comment.