search

xiongma / transformer-pointer-generator

A Abstractive Summarization Implementation with Transformer and Pointer-generator
MIT License
395 stars 79 forks source link

Does not use GPU #22

Closed ronnyli closed 5 years ago

ronnyli commented 5 years ago

(Hopefully this isn't an error on my side)

I noticed that requirements.txt only contains tensorflow==1.12.0 not tensorflow-gpu==1.12.0. Does this mean that the --gpu_nums argument for train.py doesn't do anything?

Thanks for your help!

xiongma commented 5 years ago

sorry, it my fault, it could be tensorflow-gpu==1.12.0 I will repeat it quickly

Cally Ma (马雄) Chongqing University of Post and Telecommunications Mobile: +86 15025700935 E-mail: mx15025700935@aliyun.com

发件人：ronnyli notifications@github.com 发送时间：2019年9月16日(星期一) 05:53 收件人：policeme/transformer-pointer-generator transformer-pointer-generator@noreply.github.com 抄　送：Subscribed subscribed@noreply.github.com 主　题：[policeme/transformer-pointer-generator] Does not use GPU (#22)

(Hopefully this isn't an error on my side) I noticed that requirements.txt only contains tensorflow==1.12.0 not tensorflow-gpu==1.12.0. Does this mean that the --gpu_nums argument for train.py doesn't do anything? Thanks for your help! — You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub, or mute the thread.