search

torchmd / torchmd-net

Training neural network potentials
MIT License
326 stars 73 forks source link

Disable persistent_workers with num_workers=0 and disable pin_memory in DataModule #322

Closed sef43 closed 4 months ago

sef43 commented 4 months ago

currently I cannot set num_workers to zero:

ValueError:
persistent_workers option needs num_workers > 0

This was the fix suggested by @AntonioMirarchi that works for me

Closes #323 Closes #324

RaulPPelaez commented 4 months ago

Thanks @sef43, I remember setting that option already with @AntonioMirarchi, maybe it got lost in some pending PR.