问题描述:
pycharm远程ssh连接服务器进行调试代码时,debug点击代码后pycharm卡死。
问题所在:
train_data_loader = DataLoader(train_data, batch_size=self.config.train_batch_size, sampler=RandomSampler(train_data), num_workers=1)
DataLoader中num_workers不能设置成多个。
解决方案:
把num_workers设置成0或删除即可!
train_data_loader = DataLoader(train_data, batch_size=self.config.train_batch_size, sampler=RandomSampler(train_data))