deadlock when running pretrain_reddit.py

[COK-ZhangZiliang]

Hello author, when I run this file, only one batch can be trained, and when the second batch is trained, the thread used to prepare the data is deadlocked.