RabbitRUI/ruispace

1from easydict import EasyDict as edict2 3# make training faster4# our RAM is 256G5# mount -t tmpfs -o size=140G  tmpfs /train_tmp6 7config = edict()8config.loss = "arcface"9config.network = "r50"10config.resume = False11config.output = "ms1mv3_arcface_r50"12 13config.dataset = "ms1m-retinaface-t1"14config.embedding_size = 51215config.sample_rate = 116config.fp16 = False17config.momentum = 0.918config.weight_decay = 5e-419config.batch_size = 12820config.lr = 0.1  # batch size is 51221 22if config.dataset == "emore":23    config.rec = "/train_tmp/faces_emore"24    config.num_classes = 8574225    config.num_image = 582265326    config.num_epoch = 1627    config.warmup_epoch = -128    config.decay_epoch = [8, 14, ]29    config.val_targets = ["lfw", ]30 31elif config.dataset == "ms1m-retinaface-t1":32    config.rec = "/train_tmp/ms1m-retinaface-t1"33    config.num_classes = 9343134    config.num_image = 517951035    config.num_epoch = 2536    config.warmup_epoch = -137    config.decay_epoch = [11, 17, 22]38    config.val_targets = ["lfw", "cfp_fp", "agedb_30"]39 40elif config.dataset == "glint360k":41    config.rec = "/train_tmp/glint360k"42    config.num_classes = 36023243    config.num_image = 1709165744    config.num_epoch = 2045    config.warmup_epoch = -146    config.decay_epoch = [8, 12, 15, 18]47    config.val_targets = ["lfw", "cfp_fp", "agedb_30"]48 49elif config.dataset == "webface":50    config.rec = "/train_tmp/faces_webface_112x112"51    config.num_classes = 1057252    config.num_image = "forget"53    config.num_epoch = 3454    config.warmup_epoch = -155    config.decay_epoch = [20, 28, 32]56    config.val_targets = ["lfw", "cfp_fp", "agedb_30"]57