search

eckertzhang / HumanRef

Apache License 2.0
39 stars 2 forks source link

RuntimeError: Input type (c10::Half) and bias type (float) should be the same #3

Open yc4ny opened 1 month ago

yc4ny commented 1 month ago

Thanks for the great work, I appreciate it. I'm testing this code out for resesarch purposes, but when I train the model after Initializing SDF to an mesh, on Epoch 0 I encounter this error :

python3.8/site-packages/torch/nn/modules/conv.py", line 459, in _conv_forward
    return F.conv2d(input, weight, bias, self.stride,

RuntimeError: Input type (c10::Half) and bias type (float) should be the same

Is there something that comes in mind to you which may fix the error? Thanks.