12345678910111213141516 |
- from cvargparse import Arg
- from cvfinetune.parser import default_factory
- from cvfinetune import parser as parser_module
- def parse_args():
- parser = parser_module.FineTuneParser(default_factory([
- Arg("--pretrained_on", choices=["inat", "imagenet"],
- help="network pretraining"),
- ])
- )
- return parser.parse_args()
|