Code running issues

[xxc13004]

I have installed the environment you mentioned and also downloaded the repository required for the code. But when running train_gen.sh, the following error occurred, and I couldn't find the reason why tensor calculations were performed on multiple devices in the code.

The error is as follows:

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda: 0 and CPU! (When checking argument for argument target in method wrapper_nnll_loss2d_forward)