Closed iamZe closed 4 years ago
+1
@hszhao Any reply to this? I tried training on cpu and it did not work.
Hi, currently only GPU version is supported as CPU training is much slower when compared to GPU training. To enable CPU training, you need to remove the CUDA part in 'tool/train.py' and modify the operators in 'model/san.py'.
thanks.