![](https://img-blog.csdnimg.cn/20201014180756919.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
热心市民付先生
深度学习,目标检测,目标跟踪,缺陷识别
展开
-
TADT-python报错AttributeError: ‘_tkinter.tkapp‘ object has no attribute ‘setGeometry‘
目标跟踪:TADT-pythonTarget-Aware Deep Tracking (TADT) 问题描述:AttributeError: ‘_tkinter.tkapp’ object has no attribute ‘setGeometry’错误代码详情:Traceback (most recent call last): File "demo_tadt.py", line 35, in <module> tracker.initialize_tadt(img_原创 2020-08-29 09:44:31 · 3344 阅读 · 0 评论 -
报错AttributeError: ‘torch.dtype‘ object has no attribute ‘type‘
AttributeError: ‘torch.dtype’ object has no attribute ‘type’np.mean()改为torch.mean(torch.stack(my_list))即可loss是一个tensornp.mean()输入list可以,array也可以torch.mean()必须输入tensor,最后得到的结果也是tensortorch.stack(...原创 2019-12-26 10:02:57 · 6117 阅读 · 3 评论 -
PyTorch中使用指定的GPU
PyTorch中使用指定的GPUPyTorch默认使用从0开始的GPU,如果GPU0正在运行程序,需要指定其他GPU。有如下两种方法来指定需要使用的GPU。类似tensorflow指定GPU的方式,使用CUDA_VISIBLE_DEVICES。1.1 直接终端中设定CUDA_VISIBLE_DEVICES=1 python my_script.py1.2 python代码中设定:i...原创 2019-06-11 15:23:13 · 220 阅读 · 0 评论