fix params
parent
e8a9e8f33f
commit
aaf51c7e94
|
@ -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:
|
||||||
|
|
|
@ -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:
|
||||||
|
|
|
@ -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:
|
||||||
|
|
|
@ -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:
|
||||||
|
|
|
@ -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:
|
||||||
|
|
|
@ -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:
|
||||||
|
|
|
@ -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
|
||||||
|
|
|
@ -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:
|
||||||
|
|
|
@ -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
|
||||||
|
|
||||||
|
|
|
@ -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:
|
||||||
|
|
Loading…
Reference in New Issue