Wide ResNet[1]の、Kerasでの実装です。たぶん、もっともシンプルなコードなんじゃないかと。
$ pip3 install --upgrade tensorflow-gpu keras funcy matplotlib h5py
$ python3 train.py
$ python3 check.py
私が試した結果だと、CIFAR-10の精度は95.52%になりました。論文の95.83%に近い値なので、多分コードは大丈夫。
- ごめんなさい。Python3とTensorFlowの環境でしか試していません。
- https://github.com/nutszebra/residual_netとhttps://github.com/takedarts/resnetfamilyを参考にして作成しています。
- Kerasに関数型プログラミングのテクニックを適用する方法は、Kerasと関数型プログラミングを使えば、深層学習(ディープ・ラーニング)は楽ちんですよにまとめました。
- Wide Residual Networks [1]