JohnAILove comments

Repositories
Issues
Comments

Results 1 comments of


                                            JohnAILove

My environment is 1 GPU. How do I adjust the training set?

> 嗨，我的環境也是1個GPU。您是否成功運行了培訓文件？例如train_PBAFN_stage1.py？ not , -------------- End ---------------- THCudaCheck FAIL file=/pytorch/torch/csrc/cuda/Module.cpp line=33 error=10 : invalid device ordinal Traceback (most recent call last): File "train_PBAFN_stage1.py", line 30, in torch.cuda.set_device(opt.local_rank) File "/opt/conda/lib/python3.7/site-packages/torch/cuda/__init__.py", line...