ゲームAI備忘録

ゲームAIに使えそうな知識を備忘録として書き留める

人助けと思って何卒インストールをば! 詰碁/ アルコネ/ 五目並べ

GTX1080とTITAN Xの速度比較

GTX1080とTITAN Xでの速度を比較してみました.
TensorFlowのバージョンはどちらも0.8を利用しています.
ベンチマークとして利用したのは,MSRA開発のアルゴリズムResNetの32層版です.
TensorFlow版として以前に実装したResNet32層で速度を比較してみます.

git clone https://github.com/namakemono/cifar10-tensorflow.git
cd cifar10-tensorflow
sh ./install.sh
cd cifar10
python train.py --class_name Cifar10Classifier_ResNet32
Name examples/sec
GTX 1080 2400前後
TITAN X 1600前後
Mac 35前後

おおよそ1.5倍程度の速度で学習を行えることことが分かります.

References

  • [1]. He, Kaiming, et al. "Deep Residual Learning for Image Recognition." arXiv preprint arXiv:1512.03385 (2015).

ResNetアルゴリズムの論文

TensorFlowで作成したResNetが書かれているリポジトリ