DEIM训练教程

教程有很多,我这里就不放相关链接了,主要说一下我在训练时遇到的错误和解决方案。

问题1:

envs/deim/lib/python3.11/site-packages/torchvision/transforms/v2/_transform.py", line 55, in transform
[rank0]:     raise NotImplementedError
[rank0]: NotImplementedError

可能的原因:
torch和torchvision的版本太高,对torch和torchvision的版本进行降级,下面是我运行成功的版本。

torch==2.0.1
torchvision==0.15.2

问题2:

img = torch.as_tensor(np.array(pic, copy=True,dtype=np.uint8))
RuntimeError: Could not infer dtype of numpy.uint8
或者
File "envs/DEIM/lib/python3.11/site-packages/torch/_tensor.py", line 972, in __array__
    return self.numpy().astype(dtype, copy=False)
RuntimeError: Numpy is not available

可能的原因:
numpy的版本过高和torch的版本不匹配,对numpy版本进行降级,降级到1.X的版本。

# 使用 conda(推荐)
conda activate DEIM  # 进入您的工作环境
conda install numpy

# 或使用 pip
pip install --upgrade numpy
Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐