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