diff --git a/ppcls/configs/GeneralRecognition/GeneralRecognition_PPLCNet_x2_5_binary_v2.yaml b/ppcls/configs/GeneralRecognition/GeneralRecognition_PPLCNet_x2_5_binary_v2.yaml deleted file mode 100644 index 4def782a17277f51fe07399e8f5731764a17b102..0000000000000000000000000000000000000000 --- a/ppcls/configs/GeneralRecognition/GeneralRecognition_PPLCNet_x2_5_binary_v2.yaml +++ /dev/null @@ -1,124 +0,0 @@ -# global configs -Global: - checkpoints: null - pretrained_model: null - output_dir: ./output_v3/ - device: gpu - save_interval: 1 - eval_during_train: True - eval_interval: 1 - epochs: 100 - print_batch_step: 10 - use_visualdl: False - # used for static mode and model export - image_shape: [3, 224, 224] - save_inference_dir: ./inference - eval_mode: retrieval - use_dali: False - to_static: False - - #feature postprocess - feature_normalize: False - feature_binarize: "sign" - -# model architecture -Arch: - name: RecModel - infer_output_key: features - infer_add_softmax: False - - Backbone: - name: PPLCNet_x2_5_Tanh - pretrained: True - use_ssld: True - class_num: 512 - Head: - name: CircleMargin - margin: 0.2 - scale: 30 - embedding_size: 512 - class_num: 271083 - -# loss function config for traing/eval process -Loss: - Train: - - CELoss: - weight: 1.0 - epsilon: 0.1 - Eval: - - CELoss: - weight: 1.0 - epsilon: 0.1 - -Optimizer: - name: Momentum - momentum: 0.9 - lr: - name: Cosine - learning_rate: 0.04 - warmup_epoch: 5 - regularizer: - name: 'L2' - coeff: 0.00001 - - -# data loader for train and eval -DataLoader: - Train: - dataset: - name: ImageNetDataset - image_root: ./dataset/all_data - cls_label_path: ./dataset/all_data/train_reg_all_data_v2.txt - transform_ops: - - DecodeImage: - to_rgb: True - channel_first: False - - RandCropImage: - size: 224 - - RandFlipImage: - flip_code: 1 - - NormalizeImage: - scale: 1.0/255.0 - mean: [0.485, 0.456, 0.406] - std: [0.229, 0.224, 0.225] - order: '' - - sampler: - name: DistributedBatchSampler - batch_size: 256 - drop_last: False - shuffle: True - loader: - num_workers: 4 - use_shared_memory: True - - Eval: - Query: - dataset: - name: VeriWild - image_root: ./dataset/Aliproduct/ - cls_label_path: ./dataset/Aliproduct/val_list.txt - transform_ops: - - DecodeImage: - to_rgb: True - channel_first: False - - ResizeImage: - size: 224 - - NormalizeImage: - scale: 0.00392157 - mean: [0.485, 0.456, 0.406] - std: [0.229, 0.224, 0.225] - order: '' - sampler: - name: DistributedBatchSampler - batch_size: 64 - drop_last: False - shuffle: False - loader: - num_workers: 4 - use_shared_memory: True - -Metric: - Eval: - - Recallk: - topk: [1, 5]