ASTGCN-PyTorch

Rewrite the MXNet version ASTGCN using PyTorch. Please refer to the MXNet version for more details.

ATTENTION: This repo is still under active development.

The program is runnable, but the training result is NOT correct. Issues and/or pull requests are welcome.

Requirements

• PyTorch >= 1.1.0
• SciPy

Usage

Train model on PEMS04:

    python train.py --config configurations/PEMS04.conf --force True