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()