rnn-training

Config file

Config file format

{
  "embedding_dim": 32,
  "rnn_units": 128,
  "batch_size": 512,
  "epoch_num": 30,
  "seq_length": 30
}

Running command

conda create -n env_name python=3.9
pip install -r requirements.txt
python rnn_training.py --config-path CONFIG_PATH --data-path DATASET_PATH --output-dir OUTPUT_DIR

EXAMPLE:
python rnn_training.py --data-dir ./test_training_data --config-path ./config.json

Dataset

11 datasets could be downloaded on this link rnn-char-dataset.