fix params

pull/3287/head
Sunting78 2024-11-05 09:15:50 +00:00
parent e8a9e8f33f
commit aaf51c7e94
10 changed files with 10 additions and 10 deletions

View File

@ -94,7 +94,7 @@ DataLoader:
prob: 0.5 prob: 0.5
sampler: sampler:
name: DistributedBatchSampler name: DistributedBatchSampler
batch_size: 50 batch_size: 256
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:

View File

@ -94,7 +94,7 @@ DataLoader:
prob: 0.5 prob: 0.5
sampler: sampler:
name: DistributedBatchSampler name: DistributedBatchSampler
batch_size: 50 batch_size: 256
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:

View File

@ -94,7 +94,7 @@ DataLoader:
prob: 0.5 prob: 0.5
sampler: sampler:
name: DistributedBatchSampler name: DistributedBatchSampler
batch_size: 50 batch_size: 512
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:

View File

@ -94,7 +94,7 @@ DataLoader:
prob: 0.5 prob: 0.5
sampler: sampler:
name: DistributedBatchSampler name: DistributedBatchSampler
batch_size: 50 batch_size: 512
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:

View File

@ -94,7 +94,7 @@ DataLoader:
prob: 0.5 prob: 0.5
sampler: sampler:
name: DistributedBatchSampler name: DistributedBatchSampler
batch_size: 50 batch_size: 512
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:

View File

@ -94,7 +94,7 @@ DataLoader:
prob: 0.5 prob: 0.5
sampler: sampler:
name: DistributedBatchSampler name: DistributedBatchSampler
batch_size: 50 batch_size: 512
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:

View File

@ -57,7 +57,7 @@ Optimizer:
lr: lr:
# for 8 cards # for 8 cards
name: Cosine name: Cosine
learning_rate: 0.002 ##null learning_rate: 0.003 ##null
eta_min: 1.0e-06 eta_min: 1.0e-06
warmup_epoch: 20 warmup_epoch: 20
warmup_start_lr: 0 warmup_start_lr: 0

View File

@ -111,7 +111,7 @@ DataLoader:
prob: 0.5 prob: 0.5
sampler: sampler:
name: DistributedBatchSampler name: DistributedBatchSampler
batch_size: 1024 batch_size: 512
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader:

View File

@ -30,7 +30,7 @@ AMP:
Arch: Arch:
name: MobileNetV4_hybrid_large name: MobileNetV4_hybrid_large
drop_rate: 0.2 drop_rate: 0.2
drop_path_rate: 0.1 drop_path_rate: 0.35
use_fused_attn: False use_fused_attn: False
class_num: 1000 class_num: 1000

View File

@ -109,7 +109,7 @@ DataLoader:
prob: 0.5 prob: 0.5
sampler: sampler:
name: DistributedBatchSampler name: DistributedBatchSampler
batch_size: 64 batch_size: 128
drop_last: False drop_last: False
shuffle: True shuffle: True
loader: loader: