/diin-chainer

Primary LanguagePythonApache License 2.0Apache-2.0

diin-chainer

A chainer implementation of Densely Interactive Inference Network (arXiv:1709.04348)

Requirements:

To Run,

python train.py

For debug with a small data:

python train.py --debug_mode

TODO:

  • Learning rate in adadelta
  • Use SGD in the middle of training
  • Early stopping
  • Confusion matrix
  • Use ParallelUpdater? (for data parallel processing)
  • Improve performance...