[code] PredNet in pytorch and related data

PyTorch版のPredNetをリリースしました。

Chainerライブラリーは開発が停まりましたので、深層学習の開発環境をPyTorchに移行させました。PyTorch版PredNetは、最新の環境で高速稼働します。またChainerでは不安定だった学習過程を安定させる工夫もされております。オプションも盛りだくさんです。一連の錯視の再現実験は検証済みです。バグ報告歓迎いたします。


PredNet in pytorch

https://github.com/eijwat/prednet_in_pytorch

(We recommend lr_rate option set to 0.9.)


Sample Weight Models (for Rotating Snake Illusions)

https://doi.org/10.6084/m9.figshare.16572065


Test and training data

https://ewatanabe.blogspot.com/2019/06/predictive-coding-deep-neural-networks.html




コメント

  1. お世話になります.こちらのコードを使用させていただいております.

    現在,リアルタイムで取得した画像を入力に将来の予測(拡張予測含め)を行わせたいと思い,自力でコードを書いてみましたが,通常の予測と比較したらかなり精度が落ちてしまっています.
    コードをお見せいたしますので,可能であればご助言をいただきたく存じます.

    返信削除
    返信
    1. 共有いただければ拝見させていただきます。eijwatあっとまーくgmail.comに共有してください。

      削除

コメントを投稿