search

AFeng-x / PixWizard

MIT License
123 stars 0 forks source link

Perfect work！ #2

Open zhoushen1 opened 1 month ago

zhoushen1 commented 1 month ago

Hello! Could you please tell me what kind of GPU is required for training? How many GPUs are needed? And what resource conditions are necessary to run the testing?

AFeng-x commented 1 month ago

Hi, our model requires a minimum of 8xA6000 (48GB) GPUs for training. The more and better GPUs, the faster the overall training speed will be. For inference and testing, only one V100 (32GB) GPU is needed.