Skip to content

ak110/train1000

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

49 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Train with 1000

Code style: black

訓練データを1,000件だけ使って学習してみる試み、Train with 1000をやってみたコード。

tf.kerasを使って1ファイルで全て実装したので、初心者にもわかりやすいコードになってます! (嘘)

スコア (CIFAR-10)

[INFO ] Val Accuracy:       0.8106 (5 runs)
[INFO ] Val Cross Entropy:  0.7423 (5 runs)
[INFO ] Test Accuracy:      0.8083 (5 runs)
[INFO ] Test Cross Entropy: 0.7477 (5 runs)

Valは訓練データの末尾10,000件。(独自)

動かすために必要なもの

  • TensorFlow (2.6.0で動作確認)
  • Horovod
  • OpenMPI (複数GPU時)
  • albumentations
  • scikit-learn

やってること

About

train with 1000 (http://www.ok.sc.e.titech.ac.jp/~mtanaka/proj/train1000/) をやってみたコード置き場

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages