(parser)
| 771 | |
| 772 | |
| 773 | def _add_inference_args(parser): |
| 774 | group = parser.add_argument_group(title="initialization") |
| 775 | |
| 776 | group.add_argument( |
| 777 | '--evaluation', |
| 778 | action="store_true", |
| 779 | ) |
| 780 | group.add_argument( |
| 781 | '--beam-warmup', |
| 782 | action="store_true", |
| 783 | ) |
| 784 | group.add_argument( |
| 785 | '--beam-warmup-length', |
| 786 | type=int, |
| 787 | default=0, |
| 788 | ) |
| 789 | group.add_argument( |
| 790 | '--beam-search', |
| 791 | action="store_true", |
| 792 | ) |
| 793 | group.add_argument( |
| 794 | '--beam-search-nucleus', |
| 795 | action="store_true", |
| 796 | ) |
| 797 | group.add_argument( |
| 798 | '--num-beams', |
| 799 | type=int, |
| 800 | default=4, |
| 801 | ) |
| 802 | |
| 803 | return parser |
| 804 | |
| 805 | |
| 806 | def _add_learning_rate_args(parser): |