gru_ode_bayes
gru_ode_bayes copied to clipboard
Tensors used as indices must be long, byte or bool tensors
When running climate.gruode.py, the error occurs.
p_obs = p[i_obs] IndexError: tensors used as indices must be long, byte or bool tensors
How to solve this problem?
i_obs = i_obs.to(torch.long)