Open siyu-chen-cloud opened 4 months ago
程序运行几个batch会卡住,显存被占用,但是不进行计算 环境log如下:
PyTorch built with:
很多说是互锁或者内存溢出了,但是num_works=1, batch_size=2仍然会出现卡住的情况,请问有什么办法解决?
您好,请问您是使用COCO数据集还是自定义数据集训练的模型,能否提供一下完整的训练日志以及强行中断程序后的报错信息,以方便我定位问题,谢谢。
程序运行几个batch会卡住,显存被占用,但是不进行计算 环境log如下:
PyTorch built with:
很多说是互锁或者内存溢出了,但是num_works=1, batch_size=2仍然会出现卡住的情况,请问有什么办法解决?