AttributeError: 'torch.dtype' object has no attribute 'type'
最近试一下环境配置,然后重新跑了SiamFC代码,代码链接如下:SiamFC-PyTorch
配置要求是:python2.7 python-opencv PyTorch0.4
首先处理imageNet数据集,由于没有写多线程处理,整整花了两天的时间才跑完,服务器累我心更累,由此可见多线程的重要性。处理完数据后就开始训练模型了,把batchsize 调到了64,因为128出现了GPU内存不足的报错...
原创
2019-07-15 16:33:53 ·
10572 阅读 ·
0 评论