A strange problem

[yutian-wang]

Hi,
thank you for your contribution. I test these code but occur a strange problem:

  File "test.py", line 103, in main
    _main(args)
  File "test.py", line 98, in _main
    tester.do_test()
  File "/home/wangyutian/works/buddy-master/testing/tester.py", line 233, in do_test
    self.test_dereverberation(m, blind=True)
  File "/home/wangyutian/works/buddy-master/testing/tester.py", line 153, in test_dereverberation
    pred = self.sampler.predict_conditional(y, operator_blind if blind else operator_ref, shape=(1,seg.shape[-1]), blind=blind)
  File "/home/wangyutian/works/buddy-master/testing/EulerHeunSamplerDPS.py", line 204, in predict_conditional
    return self.predict(shape, y.device, blind)
  File "/home/wangyutian/works/buddy-master/testing/EulerHeunSamplerDPS.py", line 176, in predict
    x, x_den = self.step(x, t[i] , t[i+1], gamma[i], blind)
  File "/home/wangyutian/works/buddy-master/testing/EulerHeunSamplerDPS.py", line 119, in step
    x_den = self.get_Tweedie_estimate(x_hat, t_hat)
  File "/home/wangyutian/works/buddy-master/testing/Sampler.py", line 71, in get_Tweedie_estimate
    x_hat = self.diff_params.denoiser(x.unsqueeze(1), self.model, t_i).squeeze(1)
  File "/home/wangyutian/works/buddy-master/diff_params/shared.py", line 120, in denoiser
    return cskip * xn + cout * net(cin * xn, cnoise)  #this will crash because of broadcasting problems, debug later!
  File "/usr/local/lib/python3.8/dist-packages/torch/nn/modules/module.py", line 889, in _call_impl
    result = self.forward(*input, **kwargs)
  File "/home/wangyutian/works/buddy-master/networks/ncsnpp.py", line 503, in forward
    x_spec=super().forward(x_spec, time_cond=time_cond)
  File "/home/wangyutian/works/buddy-master/networks/ncsnpp.py", line 293, in forward
    x_chans.append(torch.cat([
  File "/home/wangyutian/works/buddy-master/networks/ncsnpp.py", line 294, in <listcomp>
    torch.cat([x[:,[chan+in_chan],:,:].real, x[:,[chan+in_chan],:,:].imag ], dim=1) for in_chan in range(self.input_channels // 2)],
RuntimeError: index does not support automatic differentiation for outputs with complex dtype.

is there any solution? thank you !