1

[TUTUhappy]

你好,想问一下我将添加了这个下采样模块,但是出现了一个问题
RuntimeError: Input type (torch.cuda.HalfTensor) and weight type (torch.cuda.FloatTensor) 我仔细去检查了common 文件发现所使用的模块都放在了init 上 按道理来说应该模型跟权重应该都放到了cuda上 ,可是还是出现了这个问题,, 请问一下你有出现这个问题吗,是怎么解决的,谢谢

[pyl62112991]

你好,想问一下我将添加了这个下采样模块,但是出现了一个问题 RuntimeError: Input type (torch.cuda.HalfTensor) and weight type (torch.cuda.FloatTensor) 我仔细去检查了common 文件发现所使用的模块都放在了init 上 按道理来说应该模型跟权重应该都放到了cuda上 ,可是还是出现了这个问题,, 请问一下你有出现这个问题吗,是怎么解决的,谢谢

请问一下解决了吗

[Cpt-Lee]

你好,想问一下我将添加了这个下采样模块,但是出现了一个问题 RuntimeError: Input type (torch.cuda.HalfTensor) and weight type (torch.cuda.FloatTensor) 我仔细去检查了common 文件发现所使用的模块都放在了init 上 按道理来说应该模型跟权重应该都放到了cuda上 ,可是还是出现了这个问题,, 请问一下你有出现这个问题吗,是怎么解决的,谢谢

请问一下解决了吗

可以尝试关闭AMP，在YOLOv10里关闭AMP后可以正常训练。

[moss-code]

https://github.com/fbcotter/pytorch_wavelets/pull/51 这里面有解决AMP问题