Pin memory support in dataloader - Githubissues

a-r-r-o-w / cogvideox-factory

Memory optimized finetuning scripts for CogVideoX using TorchAO and DeepSpeed

Apache License 2.0

177 stars 16 forks source link

Pin memory support in dataloader #5

Closed a-r-r-o-w closed 6 days ago