search

gau820827 / AI-writer_Data2Doc

PyTorch Implementation of NBA game summary generator.
Apache License 2.0
83 stars 22 forks source link

Argparserwkp #41

Closed weikaipan closed 6 years ago

weikaipan commented 6 years ago

Add argparser for hyperparameter tuning.

Sample Test:

python train.py -embeddingsize 720 -learningrate 0.01 -batchsize 3 -getloss 20 -encoder HierarchicalRNN -decoder HierarchicalRNN -epochsave 12 -tocopy False -copyplayer False -gradclip 2 -maxlength 12 -maxsentece 800