wireless-intelligence-lab / DeepLearning-CoordinatedBeamforming

Simulation code for "Deep Learning Coordinated Beamforming for Highly-Mobile Millimeter Wave Systems" by Ahmed Alkhateeb, Sam Alex, Paul Varkey, Ying Li, Qi Qu, and Djordje Tujkovic, in IEEE Access, vol. 6, pp. 37328-37348, 2018.
https://arxiv.org/abs/1804.10334
Other
59 stars 25 forks source link

How to modify CPU training instead of GPU in DLmodel_python_code.py file? #1

Closed fc-smile2 closed 2 years ago

fc-smile2 commented 5 years ago

In the case of data set applications, ”Deep Learning Coordinated Beamforming for Highly-Mobile Millimeter Wave Systems” is trained with GPU in ‘DL model_python_code.py file', but my computer does not have GPU, so how should I modify this file to use CPU training?

yaling1230 commented 4 years ago

并不需要更改什么,只需要将"os.environ["THEANO_FLAGS"] = "device=gpu%d"%(1)"这行注释掉就可以使用了,但CPU训练较慢,推荐使用GPU。