DEIM训练教程
·
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
更多推荐


所有评论(0)