parser.py 326 B

12345678910111213141516
  1. from cvargparse import Arg
  2. from cvfinetune.parser import default_factory
  3. from cvfinetune import parser as parser_module
  4. def parse_args():
  5. parser = parser_module.FineTuneParser(default_factory([
  6. Arg("--pretrained_on", choices=["inat", "imagenet"],
  7. help="network pretraining"),
  8. ])
  9. )
  10. return parser.parse_args()