search

zsyOAOA / DifFace

DifFace: Blind Face Restoration with Diffused Error Contraction (TPAMI, 2024)
Other
628 stars 42 forks source link

It's not convenient to change the cuda device when running inferrence script. #22

Open thu-huangzc opened 5 months ago

thu-huangzc commented 5 months ago

Could you add a device number to the command line parameters?

zsyOAOA commented 5 months ago

Please use CUDA_VISIBLE_DEVICES directly.