diff --git a/ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_small_x1_0_ultra.yaml b/ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_small_x1_0_ultra.yaml index 08c0b5119dda3ff81126e87764065f3d8086bae7..e31358717458211c077764336f7151917d5f6c66 100644 --- a/ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_small_x1_0_ultra.yaml +++ b/ppcls/configs/ImageNet/MobileNetV3/MobileNetV3_small_x1_0_ultra.yaml @@ -43,7 +43,7 @@ Optimizer: momentum: 0.9 lr: name: Cosine - learning_rate: 7.8 + learning_rate: 5.2 warmup_epoch: 5 regularizer: name: 'L2' @@ -73,11 +73,11 @@ DataLoader: sampler: name: DistributedBatchSampler - batch_size: 1536 + batch_size: 1024 drop_last: False shuffle: True loader: - num_workers: 20 + num_workers: 16 use_shared_memory: True Eval: diff --git a/test_tipc/configs/MobileNetV3/MobileNetV3_small_x1_0_ultra_train_infer_python.txt b/test_tipc/configs/MobileNetV3/MobileNetV3_small_x1_0_ultra_train_infer_python.txt index ce429db32f353f5e14d4a8eb5987b3e3aad9c1a4..4f7c56919e7fe09bbed9067d15e5fb17f9f4e8b4 100644 --- a/test_tipc/configs/MobileNetV3/MobileNetV3_small_x1_0_ultra_train_infer_python.txt +++ b/test_tipc/configs/MobileNetV3/MobileNetV3_small_x1_0_ultra_train_infer_python.txt @@ -50,11 +50,11 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o Global.benchmark:False null:null ===========================train_benchmark_params========================== -batch_size:1536 +batch_size:1024 fp_items:fp16 epoch:1 model_type:norm_train -num_workers:20 +num_workers:16 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile flags:FLAGS_eager_delete_tensor_gb=0.0;FLAGS_fraction_of_gpu_memory_to_use=0.98;FLAGS_conv_workspace_size_limit=4096 ===========================infer_benchmark_params==========================