RuntimeError: Expected object of scalar type Double but got scalar type Float for argument #2 'weight'

[andkobe]

你好，我在跑test_image.py时遇到这个问题 RuntimeError: Expected object of scalar type Double but got scalar type Float for argument #2 'weight'， 打印如下： Traceback (most recent call last): File "E:/Code/MachineLearning/DFace-master/test_image.py", line 18, in bboxs, landmarks = mtcnn_detector.detect_face(img) File "E:\Code\MachineLearning\DFace-master\dface\core\detect.py", line 610, in detect_face boxes, boxes_align = self.detect_pnet(img) File "E:\Code\MachineLearning\DFace-master\dface\core\detect.py", line 271, in detect_pnet cls_map, reg = self.pnet_detector(feed_imgs) File "D:\ProgramData\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 547, in call result = self.forward(*input, **kwargs) File "E:\Code\MachineLearning\DFace-master\dface\core\models.py", line 97, in forward x = self.pre_layer(x) File "D:\ProgramData\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 547, in call result = self.forward(*input, **kwargs) File "D:\ProgramData\Anaconda3\lib\site-packages\torch\nn\modules\container.py", line 92, in forward input = module(input) File "D:\ProgramData\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 547, in call result = self.forward(*input, **kwargs) File "D:\ProgramData\Anaconda3\lib\site-packages\torch\nn\modules\conv.py", line 343, in forward return self.conv2d_forward(input, self.weight) File "D:\ProgramData\Anaconda3\lib\site-packages\torch\nn\modules\conv.py", line 340, in conv2d_forward self.padding, self.dilation, self.groups) RuntimeError: Expected object of scalar type Double but got scalar type Float for argument #2 'weight'

Process finished with exit code 1

有朋友遇到类似问题么？感谢！

[SimonFFL]

将./dface/core/image_tools.py文件中第20行改为： return (transform(image) / 255).float()

[kuaikuaikim]

这是来自QQ邮箱的假期自动回复邮件。你好，我最近正在休假中，无法亲自回复你的邮件。我将在假期结束后，尽快给你回复。