iSEE-Laboratory / DiffUIR

The official implementation of the paper of CVPR2024: Selective Hourglass Mapping for Universal Image Restoration Based on Diffusion Model
MIT License
103 stars 7 forks source link

咨询训练相关的问题 #15

Closed jjjjjgggj closed 2 months ago

jjjjjgggj commented 2 months ago

源代码是在cpu上训练嘛

zhengdian1 commented 2 months ago

肯定是gpu啊,4090上训练的