运行test.py是会卡死，但是单步运行不报错。

[lxq0204]

运行sh test.sh时，当运行到compute_on_dataset函数的for循环（for i,batch in enumerate(tqdm(data_loader))） 时程序卡死，但不会报错。 大家有遇到这个问题的吗？