Persistent worker Error

I am getting the error
TypeError: init() got an unexpected keyword argument ‘persistent_workers’
I am using torch ==1.5.1
It doesn’t matter even if I makes pin memory False.

Your PyTorch version might be too old, so update to the latest stable or nightly release and this argument should work.