Computer Vision & Machine Learning Research Laboratory
torch.nan_to_num
This is for 1.8 version.
so use it instead of this.
temp[torch.isnan(temp)] = 0
temp is tensor
No comments:
Post a Comment