Kyrie10favor comments

Repositories
Issues
Comments

Results 1 comments of


                                            Kyrie10favor

When I set CUDA_VISIBLE_DEVICES="1" not "0"，The Code dose not work well.

I solved it this way. **Original code** ``` self.model.to(self.device) ``` **New Code** ``` device_ids = [4,5,6,7] self.device = device_ids[0] self.model.to(self.device) ```