yang-stephen

关于单机单卡遇上的报错

3

你好，非常想知道单卡单机训练时该怎么修改代码，自己尝试直接单卡运行，在此处报错 Traceback (most recent call last): File "train.py", line 429, in train() File "train.py", line 292, in train out, out16, out32, detail8 = net(im) File "/root/miniconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl...