diff --git a/README.md b/README.md index b227e40..9b526c3 100644 --- a/README.md +++ b/README.md @@ -3,6 +3,7 @@ [![python-url](https://img.shields.io/badge/Python-3.x-red.svg)](https://www.python.org/) [![pytorch-url](https://img.shields.io/badge/Pytorch>=1.4-blue.svg)](https://pytorch.org/) [![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](LICENSE) +![Docker Pulls](https://img.shields.io/docker/pulls/cavall/cavaface_env?logo=docker). By Yaobin Li and Liying Chi diff --git a/config.py b/config.py index bd8493c..4875529 100644 --- a/config.py +++ b/config.py @@ -2,27 +2,27 @@ 1: dict( SEED=1337, SYNC_DATA=True, - SYNC_DATA_NUMCLASS=50 * 10000, # 2059904, + SYNC_DATA_NUMCLASS=100 * 10000, # 2059904, DATA_ROOT="../data/webface260m", VAL_SET="lfw, cfp_fp, agedb_30", MODEL_ROOT="../models/tmp/model", IS_RESUME=False, BACKBONE_RESUME_ROOT="", HEAD_RESUME_ROOT="", - BACKBONE_NAME="IR_50", # support: ['MobileFaceNet', 'ResNet_50', 'ResNet_101', 'ResNet_152', + BACKBONE_NAME="IR_100", # support: ['MobileFaceNet', 'ResNet_50', 'ResNet_101', 'ResNet_152', #'IR_50', 'IR_101', 'IR_152', 'IR_SE_50', 'IR_SE_100', 'IR_SE_101', 'IR_SE_152', #'AttentionNet_IR_56', 'AttentionNet_IRSE_56','AttentionNet_IR_92', 'AttentionNet_IRSE_92', #'ResNeSt_50', 'ResNeSt_101', 'ResNeSt_100'] - HEAD_NAME="ArcFace", # ArcFace, ParallelArcFace - LOSS_NAME="CrossEntropy", # CrossEntropy, ParallelArcLoss - ENABLE_MODEL_PARALLEL=False, + HEAD_NAME="ParallelArcFace", # ArcFace, ParallelArcFace + LOSS_NAME="ParallelArcLoss", # CrossEntropy, ParallelArcLoss + ENABLE_MODEL_PARALLEL=True, MODEL_PARALLEL_DIM=1, # -1/1/0 ENABLE_AMP=True, OPTIMIZER="sgd", - ENABLE_ZERO_OPTIM=False, + ENABLE_ZERO_OPTIM=True, INPUT_SIZE=[112, 112], EMBEDDING_SIZE=512, - BATCH_SIZE=1024, + BATCH_SIZE=512, EVAL_FREQ=2000, LR=0.1, LR_SCHEDULER="cosine", diff --git a/docs/INSTALL.md b/docs/INSTALL.md index 690eaad..3393ee5 100644 --- a/docs/INSTALL.md +++ b/docs/INSTALL.md @@ -1,13 +1,3 @@ # Installation -## Requirements - -1. Install pytorch==1.4.0 and torchvision==0.5.0 following [official instruction](https://pytorch.org/). -2. Clone this repo. -3. Install dependencies: - - ```python - pip install -r requirements.txt - ``` - -4. (Option) If you are used to using docker, the dockerhub is [pytorch_env](https://hub.docker.com/r/cavall/pytorch_env) ![Docker Pulls](https://img.shields.io/docker/pulls/cavall/pytorch_env?logo=docker). \ No newline at end of file +Please pull docker images [cavaface_env](https://hub.docker.com/r/cavall/cavaface_env) \ No newline at end of file diff --git a/train.py b/train.py index fe747d5..d3cc2ec 100644 --- a/train.py +++ b/train.py @@ -175,7 +175,6 @@ def print_pass(*args): {"params": backbone_paras_only_bn}, ] if cfg["ENABLE_ZERO_OPTIM"]: - print("using ZeroRedundancyOptimizer") optimizer = ZeroRedundancyOptimizer( backbone.parameters(), optimizer_class=torch.optim.SGD,