Is there any way to convert saved weight data type from torch.FloatTensor to torch.HalfTensor

[JetBlock]

I trained the model in Dtype torch.cuda.FloatTensor, now I want to load the checkpoint and test in Dtype torch.cuda.HalfTensor.
Must I train the model again? or how can I conver checkpoint weight’s Dtype from float32 to float16

[Pranavan_Theivendira (Pranavan Theivendiram)]

Hi,

Similar question has been asked and discussed previously in these forums. You can have a look here.

How to cast a tensor to another type?

if I got a float tensor,but the model needs a double tensor.what should I do to cast the float tensor to double tensor?

Thanks

[JetBlock]

thank you for your reply, I have solved this question by your link.