!32617 修复 ST probabilistic failure in daily version

Merge pull request !32617 from 王程浩/master
This commit is contained in:
i-robot 2022-04-13 06:37:02 +00:00 committed by Gitee
commit d1bf439718
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 3 additions and 3 deletions

View File

@ -19,7 +19,7 @@ from easydict import EasyDict as ed
config = ed({ config = ed({
"class_num": 1001, "class_num": 1001,
"batch_size": 32, "batch_size": 64,
"loss_scale": 128, "loss_scale": 128,
"momentum": 0.9, "momentum": 0.9,
"weight_decay": 5e-4, "weight_decay": 5e-4,
@ -33,7 +33,7 @@ config = ed({
"save_checkpoint_path": "./", "save_checkpoint_path": "./",
"label_smooth": 1, "label_smooth": 1,
"label_smooth_factor": 0.1, "label_smooth_factor": 0.1,
"frequency": 834, "frequency": 2048,
"eval_interval": 1, "eval_interval": 1,
"eval_batch_size": 32, "eval_batch_size": 32,
"train_image_size": 224, "train_image_size": 224,

View File

@ -339,7 +339,7 @@ def thor_end(device_num, q):
assert thor_cost < 30 assert thor_cost < 30
@pytest.mark.level1 @pytest.mark.level0
@pytest.mark.platform_arm_ascend_training @pytest.mark.platform_arm_ascend_training
@pytest.mark.platform_x86_ascend_training @pytest.mark.platform_x86_ascend_training
@pytest.mark.env_single @pytest.mark.env_single