Skip to content

coltenfowler83/MachineTranslation

Repository files navigation

Encoder / Decoder hidden size are setable command line args, but not required.  If they are not set at cmd line, they
will default to encoder_hidden_size=64, decoder_hidden_size=128

Attention mode required command line arguments, defaults to "mult" for multiplicative att,
use "add" for additive, use "dot" for dot product attention (e.g. "--attention_mode" "dot")

Arch is added as an additional required command line argument.  This specifies whether to use an LSTM or GRU architecture.
Set as an argument using "--arch" "LSTM" or "--arch" "GRU"

Vocab generation unchanged.

To run main in train mode, an example set of arguments is:
"--mode" "train" "--model_path" "model.bin" "--attention_mode" "dot" "--arch" "LSTM"

To run main in eval mode, example arguments:
"--arch" "LSTM" "--mode" "eval_test" "--model_path" "model.bin" "--attention_mode" "dot" "--output_file" "test.out"

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published