pytorch运行卡住,即不报错也不停止

程序运行到加载数据时卡住

for _, (support, query) in enumerate(eval_dataset)

把dataloader里的num_workers=0也没用,但是把eunmerate去掉有用了,但是运行到一半又停下了

screen 还有影响?
一开始卡着,screen界面没有输出,我ctrl+a+[ 滚动浏览screen界面,按下Esc 退出浏览,screen界面又有输出了,又能继续执行了。

数据集太大
train_dataset不慢,eval_dataset特别慢,可能是数据集太大train_dataset占据了gpu的cache?反正数据调小就快多了

你可能感兴趣的:(pytorch,人工智能,python)