Stabilize full SFT
parent
fa7dff8291
commit
41cacac703
|
|
@ -185,7 +185,7 @@ if __name__ == "__main__":
|
|||
batch_size=args.batch_size,
|
||||
pin_memory=True,
|
||||
drop_last=False,
|
||||
shuffle=False,
|
||||
shuffle=True,
|
||||
num_workers=args.num_workers,
|
||||
sampler=train_sampler
|
||||
)
|
||||
|
|
|
|||
Loading…
Reference in New Issue