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.