現在2400枚(testデータ:600枚,trainデータ:2400枚)の画像データを中間層1つ(ノード約600)のニューラルネットワークを作成しています。
条件としては
epoch 1回
batch_size 100
隠れ層 600
学習率 0.01
のミニバッチ学習をしています。
ジュピターのノートブックでこのネットワークを回すと、途中でとまってしまいます。
ちなみにbatch_sizeを大きくしたり、隠れ層のノードを少なくしたりすると、結果がしっかりと帰ってくるので、プログラムコードはあっていると思います。
そこで、google colaboratory を用いてみましたが、途中でランタイムとなり、止まってしまいました。
私のプログラムコードは、ネットから拾ってきたとても一般的なニューラルネットワークですが、皆さんはどのような環境で深層学習をしているのか教えていただけないでしょうか。
初歩的な質問かとは思いますがよろしくお願いします。
あなたの回答
tips
プレビュー