Skip to content

Commit

Permalink
[Fix] fix configs (#783)
Browse files Browse the repository at this point in the history
fix configs
  • Loading branch information
HIT-cwh authored Jun 24, 2024
1 parent 937e993 commit 8c3b44d
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 5 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -117,7 +117,7 @@
batch_size=batch_size,
num_workers=dataloader_num_workers,
dataset=alpaca_en,
sampler=dict(type=sampler, shuffle=True, seed=1024),
sampler=dict(type=sampler, shuffle=True),
collate_fn=dict(type=default_collate_fn, use_varlen_attn=use_varlen_attn))

#######################################################################
Expand Down Expand Up @@ -185,8 +185,7 @@
checkpoint=dict(
type=CheckpointHook,
by_epoch=False,
interval=-1,
save_last=False,
interval=save_steps,
max_keep_ckpts=save_total_limit),
# set sampler seed in distributed evrionment.
sampler_seed=dict(type=DistSamplerSeedHook),
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -184,8 +184,7 @@
checkpoint=dict(
type=CheckpointHook,
by_epoch=False,
interval=-1,
save_last=False,
interval=save_steps,
max_keep_ckpts=save_total_limit),
# set sampler seed in distributed evrionment.
sampler_seed=dict(type=DistSamplerSeedHook),
Expand Down

0 comments on commit 8c3b44d

Please sign in to comment.