Attribute Error when resuming training from checkpoint file

[pamparana34]

I am having a strange issue when resuming training with a ckpt_file. While the model seems to train ok, I get this error which gets printed continuously on the screen:

if not self._shutdown:
AttributeError: '_MultiProcessingDataLoaderIter' object has no attribute '_shutdown'
    self._shutdown_workers()

Not sure if this is a known issue or where do I need to look to start fixing this...