Pytorch ELM
Pytorch implementation of Extreme Learning Machines.
Prerequisites
- Python 3.3+
- Pytorch
Usage
This code is implemented for MNIST dataset.
Usage example:
$ python train_elm.py --hsize 1000
--hsize represents the number of neurons in hidden layer.