Eun Yeong Jeon

icon location Seoul,Korea

Results 1 comments of Eun Yeong Jeon

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument index in method wrapper__index_select)`

> @xzdong-2019, may I ask how did you solve that? I mean where should us add gpus=1? `python main.py --gpus 0, --prompt ....` it works to me