import flops_counter
import verification
sys.path.append(os.path.join(os.path.dirname(__file__), '..', 'symbol'))
import fresnet
import fmobilefacenet
import fmobilenet
import fmnasnet
import fdensenet
import vargfacenet
from util import CommonUtil as cu
import logging
logger = logging.getLogger()
logger.setLevel(logging.INFO)
cu.setup_logger(logging, 'train-ilfw.log')
#logging.basicConfig(filename='train-ilfw.log', level=logging.INFO)
args = None
def parse_args():
parser = argparse.ArgumentParser(
description='Insightface Ilfw Model Training Script')
# general
parser.add_argument('--dataset',
default=default.dataset,
help='dataset config')
parser.add_argument('--network',
default=default.network,
help='network config')
