fix params
parent
e8a9e8f33f
commit
aaf51c7e94
|
@ -94,7 +94,7 @@ DataLoader:
|
|||
prob: 0.5
|
||||
sampler:
|
||||
name: DistributedBatchSampler
|
||||
batch_size: 50
|
||||
batch_size: 256
|
||||
drop_last: False
|
||||
shuffle: True
|
||||
loader:
|
||||
|
|
|
@ -94,7 +94,7 @@ DataLoader:
|
|||
prob: 0.5
|
||||
sampler:
|
||||
name: DistributedBatchSampler
|
||||
batch_size: 50
|
||||
batch_size: 256
|
||||
drop_last: False
|
||||
shuffle: True
|
||||
loader:
|
||||
|
|
|
@ -94,7 +94,7 @@ DataLoader:
|
|||
prob: 0.5
|
||||
sampler:
|
||||
name: DistributedBatchSampler
|
||||
batch_size: 50
|
||||
batch_size: 512
|
||||
drop_last: False
|
||||
shuffle: True
|
||||
loader:
|
||||
|
|
|
@ -94,7 +94,7 @@ DataLoader:
|
|||
prob: 0.5
|
||||
sampler:
|
||||
name: DistributedBatchSampler
|
||||
batch_size: 50
|
||||
batch_size: 512
|
||||
drop_last: False
|
||||
shuffle: True
|
||||
loader:
|
||||
|
|
|
@ -94,7 +94,7 @@ DataLoader:
|
|||
prob: 0.5
|
||||
sampler:
|
||||
name: DistributedBatchSampler
|
||||
batch_size: 50
|
||||
batch_size: 512
|
||||
drop_last: False
|
||||
shuffle: True
|
||||
loader:
|
||||
|
|
|
@ -94,7 +94,7 @@ DataLoader:
|
|||
prob: 0.5
|
||||
sampler:
|
||||
name: DistributedBatchSampler
|
||||
batch_size: 50
|
||||
batch_size: 512
|
||||
drop_last: False
|
||||
shuffle: True
|
||||
loader:
|
||||
|
|
|
@ -57,7 +57,7 @@ Optimizer:
|
|||
lr:
|
||||
# for 8 cards
|
||||
name: Cosine
|
||||
learning_rate: 0.002 ##null
|
||||
learning_rate: 0.003 ##null
|
||||
eta_min: 1.0e-06
|
||||
warmup_epoch: 20
|
||||
warmup_start_lr: 0
|
||||
|
|
|
@ -111,7 +111,7 @@ DataLoader:
|
|||
prob: 0.5
|
||||
sampler:
|
||||
name: DistributedBatchSampler
|
||||
batch_size: 1024
|
||||
batch_size: 512
|
||||
drop_last: False
|
||||
shuffle: True
|
||||
loader:
|
||||
|
|
|
@ -30,7 +30,7 @@ AMP:
|
|||
Arch:
|
||||
name: MobileNetV4_hybrid_large
|
||||
drop_rate: 0.2
|
||||
drop_path_rate: 0.1
|
||||
drop_path_rate: 0.35
|
||||
use_fused_attn: False
|
||||
class_num: 1000
|
||||
|
||||
|
|
|
@ -109,7 +109,7 @@ DataLoader:
|
|||
prob: 0.5
|
||||
sampler:
|
||||
name: DistributedBatchSampler
|
||||
batch_size: 64
|
||||
batch_size: 128
|
||||
drop_last: False
|
||||
shuffle: True
|
||||
loader:
|
||||
|
|
Loading…
Reference in New Issue