Arguments for BMInf
(parser)
| 12 | |
| 13 | |
| 14 | def add_bminf_args(parser): |
| 15 | """Arguments for BMInf""" |
| 16 | group = parser.add_argument_group("BMInf") |
| 17 | |
| 18 | group.add_argument("--bminf", action="store_true", help="Use BMInf to support low resource evaluation") |
| 19 | group.add_argument("--bminf-memory-limit", type=int, default=20, help="Max memory for model per GPU (in GB)") |
| 20 | return parser |
| 21 | |
| 22 | |
| 23 | def add_quantization_args(parser): |