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