mattrobin:

TypeError: can’t pickle _thread.lock objects


I run into a similar issue in DDP where I loaded tfrecord dataset with more than 1 number_workds and I got the same error. TypeError: can’t pickle _thread.lock objects.
Any update? Thanks in advance!