New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

在运行处理训练数据5.py文件的时出现ByteTensor和FloatTensor不统一的问题 #19

Open

junasir opened this issue Feb 15, 2021 · 0 comments

junasir commented Feb 15, 2021

问题如下
RuntimeError: Input type (torch.cuda.ByteTensor) and weight type (torch.cuda.FloatTensor) should be the same

解决方法：
在 _,out = resnet101(img2)这一行前面加上img2=img2.float()，一共有2出

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment