Open jiangduwang opened 1 year ago
输出Epoch之后就不动了,nvidia-smi看也没有吃显卡资源。
解决了,原来是pytorch版本问题,从1.5.1改为1.10.1就行了
输出Epoch之后就不动了,nvidia-smi看也没有吃显卡资源。