mirror of
https://github.com/PaddlePaddle/PaddleOCR.git
synced 2025-06-03 21:53:39 +08:00
32 lines
444 B
YAML
32 lines
444 B
YAML
|
|
||
|
|
class_dim: 100
|
||
|
|
total_images: 50000
|
||
|
|
epochs: 1000
|
||
|
|
topk: 5
|
||
|
|
save_model_dir: ./output/
|
||
|
|
use_gpu: True
|
||
|
|
model_type: cls_distill
|
||
|
|
|
||
|
|
LEARNING_RATE:
|
||
|
|
function: 'Cosine'
|
||
|
|
params:
|
||
|
|
lr: 0.001
|
||
|
|
warmup_epoch: 5
|
||
|
|
|
||
|
|
OPTIMIZER:
|
||
|
|
function: 'Momentum'
|
||
|
|
params:
|
||
|
|
momentum: 0.9
|
||
|
|
regularizer:
|
||
|
|
function: 'L2'
|
||
|
|
factor: 0.00002
|
||
|
|
|
||
|
|
TRAIN:
|
||
|
|
batch_size: 1280
|
||
|
|
num_workers: 4
|
||
|
|
|
||
|
|
VALID:
|
||
|
|
batch_size: 64
|
||
|
|
num_workers: 4
|
||
|
|
|