Open faiaz2011 opened 1 year ago
When I try to train using a custom dataset, I get the following error.
RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.cuda.HalfTensor) should be the same.
Does anybody face a similar issue? any solution regarding this?
我也是,您解决了吗?
No.
When I try to train using a custom dataset, I get the following error.
RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.cuda.HalfTensor) should be the same.
Does anybody face a similar issue? any solution regarding this?