-
Notifications
You must be signed in to change notification settings - Fork 0
coltenfowler83/MachineTranslation
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
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 0
No packages published