Examples and descriptions are in description.ipynb To run the notebook and data_generator need to install requirements.txt which basically includes numpy and matplotlib.
To run codes relates to the model need to install requirements_full.txt which includes pytorch and pytorch lightning.
For database generation, use database_generator.py. The parameters of the database are defined in batch_gen_util.py. A memory calculator function to calculate the memory given a kernel is included. All parameters and memory information will be written to a .txt file.
seq2seq-data-generator is at the same level w.r.t. 3W.