error when run demo

[hepytobecool]

Testing gpu [0, 1, 2, 3, 4, 5, 6] Network [RotateSPADEGenerator] was created. Total number of parameters: 225.1 million. To see the architecture, do print(network). start prefetching data... Process Process-1: Traceback (most recent call last): File "/home/yangchuqiao/.conda/envs/render/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap self.run() File "/home/yangchuqiao/.conda/envs/render/lib/python3.8/multiprocessing/process.py", line 108, in run self._target(*self._args, **self._kwargs) File "/home/yangchuqiao/code/github/Rotate-and-Render/data/data_utils.py", line 146, in prefetch_data prefetcher = data_prefetcher(dataloader, opt, render_layer) File "/home/yangchuqiao/code/github/Rotate-and-Render/data/data_utils.py", line 99, in init self.preload() File "/home/yangchuqiao/code/github/Rotate-and-Render/data/data_utils.py", line 124, in preload self.next_input = get_multipose_test_input(data, self.render_layer, self.opt.yaw_poses, self.opt.pitch_poses) File "/home/yangchuqiao/code/github/Rotate-and-Render/data/data_utils.py", line 65, in get_multipose_test_input = render.rotate_render(data['param_path'], real_image, data['M'], yaw_pose=pose) File "/home/yangchuqiao/code/github/Rotate-and-Render/models/networks/rotate_render.py", line 100, in rotate_render tex_a_crop = self.get_render_from_vertices(rendered_images_erode[n], vertices_in_ori_img[n]) File "/home/yangchuqiao/code/github/Rotate-and-Render/models/networks/render.py", line 340, in get_render_from_vertices textures = img_ori[vertices_in_ori_img[1, :].round().clamp(0, h - 1).long(), \ RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)

[lya001]

I ran into the same problem and solved it by adding this before line 340: vertices_in_ori_img = vertices_in_ori_img.to(device="cpu")