AssertionError: If using tupled rtol it must have the same length as the tuple y0

[xiaoxiaoxiaogu]

In training, I encountered some problems ：
Traceback (most recent call last):
File "/home/gukaifeng/桌面/StyleFlow-master/train_flow.py", line 97, in
loss.backward()
File "/home/gukaifeng/anaconda3/lib/python3.9/site-packages/torch/_tensor.py", line 363, in backward
torch.autograd.backward(self, gradient, retain_graph, create_graph, inputs=inputs)
File "/home/gukaifeng/anaconda3/lib/python3.9/site-packages/torch/autograd/init.py", line 173, in backward
Variable._execution_engine.run_backward( # Calls into the C++ engine to run the backward pass
File "/home/gukaifeng/anaconda3/lib/python3.9/site-packages/torch/autograd/function.py", line 253, in apply
return user_fn(self, *args)
File "/home/gukaifeng/anaconda3/lib/python3.9/site-packages/torchdiffeq/_impl/adjoint.py", line 126, in backward
aug_state = odeint(
File "/home/gukaifeng/anaconda3/lib/python3.9/site-packages/torchdiffeq/_impl/odeint.py", line 72, in odeint
shapes, func, y0, t, rtol, atol, method, options, event_fn, t_is_reversed = _check_inputs(func, y0, t, rtol, atol, method, options, event_fn, SOLVERS)
File "/home/gukaifeng/anaconda3/lib/python3.9/site-packages/torchdiffeq/_impl/misc.py", line 207, in _check_inputs
rtol = _tuple_tol('rtol', rtol, shapes)
File "/home/gukaifeng/anaconda3/lib/python3.9/site-packages/torchdiffeq/_impl/misc.py", line 115, in _tuple_tol
assert len(tol) == len(shapes), "If using tupled {} it must have the same length as the tuple y0".format(name)
AssertionError: If using tupled rtol it must have the same length as the tuple y0

[xiaoxiaoxiaogu]

I tried the training dataset you gave ，but it still reported the error

[wmpscc]

I have the same question. But I solved it by reinstalling torchdiffeq==0.0.1.