Skip to content

Commit

Permalink
fix vocab size for debugmodel and real data
Browse files Browse the repository at this point in the history
  • Loading branch information
samsja committed Sep 28, 2024
1 parent 879828a commit a64ef88
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions src/zeroband/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -92,6 +92,7 @@ def train(config: Config):
num_workers=config.data.num_workers,
fake_data=config.data.fake,
)

model, model_config = get_model(
config.name_model,
config.type_model,
Expand Down

0 comments on commit a64ef88

Please sign in to comment.