futr3d
futr3d copied to clipboard
pytorch1.6has no module named nan_to_num
nan_to_num was released in pytorch1.8+, so there got an error in the code, please check and fix it
torch. nan_to_num should be replaced with numpy. nan_to_num