Skip to content

Latest commit

 

History

History

t2t_paper_generation_problem

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
PaperGenerationProblem.py
PaperGenerationProblem.py
 
 
README.md
README.md
 
 
__init__.py
__init__.py
 
 
test_paper_gen_problem.py
test_paper_gen_problem.py
 
 
train.py
train.py
 
 

README.md

Tensor2Tensor Paper Generation Problem

Install Dependecies

pip install tensor2tensor==1.13.2

Train Model

First, you need to change PaperGenerationProblem.py line 20 with the URL to the dataset created using the code specied in the main README.

python t2t_paper_generation_problem/train.py

usage: train.py [-h] [--folder FOLDER] [--model MODEL]
                [--hparams_set HPARAMS_SET]

optional arguments:
  -h, --help            show this help message and exit
  --folder FOLDER
  --model MODEL
  --hparams_set HPARAMS_SET

Generate Paper from the Trained Model

See Tensor2Tensor