KarthikR:

adaptive = robust_loss_pytorch.adaptive.AdaptiveLossFunction(num_dims = 4, float_dtype= torch.float32 , device=“cuda:0”)


Thanks sir! It does work.However,what I am wondering is that all of my data and model is in cuda, if I use torch.float32, isn’t  it a mismatch of data type?
Again,Thanks for such an caring responses : )
You have a nice day, sir!