Scripts for automatically generating hpm files for Neural Machine Translation models trained with Sockeye given possible values of hyperparameters.
-
Specify
workdir
,rootdir
,src
,trg
, etc. in*.hpm.template
files. Notice???
is a placeholder that will be replaced later with actual values from*_hpm_dict.py
. -
Specify hyperparameter values in
*_hpm_dict.py
. -
Make a directory for generated hpm files, e.g.
rnn_hpm
. -
Run scripts to generate hpm files: RNN models:
python hpm_generator.py -a rnn -o rnn_hpm
CNN models:
python hpm_generator.py -a cnn cnn_hpm
Transformer models:
python hpm_generator.py -a trans trans_hpm