16 batch_size 3886MB AutoModel .from_pretrained(MODEL_NAME) 执行慢,原因是需要启用网络代理,否则总是卡在验证阶段 DataLoader 增加num_workers后 torch.cuda.OutOfMemoryError: CUDA out of memory. 原因可能是把import放在了if __name__ == '__main__':外边?或者我更新了一下显卡驱动? 神奇了,过了一天...