Skip to content

Commit

Permalink
Update README.md for Argument (pytorch#711)
Browse files Browse the repository at this point in the history
  • Loading branch information
soumith authored Mar 5, 2020
1 parent db0dbf6 commit 6f38987
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions word_language_model/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -43,6 +43,7 @@ optional arguments:
--cuda use CUDA
--log-interval N report interval
--save SAVE path to save the final model
--onnx-export path to export the final model in onnx format
--transformer_head N the number of heads in the encoder/decoder of the transformer model
--transformer_encoder_layers N the number of layers in the encoder of the transformer model
--transformer_decoder_layers N the number of layers in the decoder of the transformer model
Expand Down

0 comments on commit 6f38987

Please sign in to comment.