search

masayoshi-ogura / HDNNP

high dimensional neural network potential
MIT License
19 stars 8 forks source link

chainerMN #131

Closed masayoshi-ogura closed 6 years ago

masayoshi-ogura commented 6 years ago

分散深層学習パッケージchainerMNを導入してみる chainerの学習部分を並列化できると期待

http://chainermn.readthedocs.io/en/stable/installation/guide.html

masayoshi-ogura commented 6 years ago

cross validation時には学習が1コアで進むように、modules/hdnnp.py内のmpi.chainer_commの部分をrun()の引数に入れてmode=cvとmode=trainingで違う挙動になるようにする