62 lines
1.4 KiB
YAML
62 lines
1.4 KiB
YAML
GPUID: 0
|
|
WORKERS: 1
|
|
PRINT_FREQ: 10
|
|
SAVE_FREQ: 10
|
|
PIN_MEMORY: False
|
|
OUTPUT_DIR: 'output'
|
|
|
|
CUDNN:
|
|
BENCHMARK: True
|
|
DETERMINISTIC: False
|
|
ENABLED: True
|
|
|
|
DATASET:
|
|
DATASET: 360CC
|
|
ROOT: "../textGenerator/dataset/dataset9/images"
|
|
CHAR_FILE: '../textGenerator/dataset/dataset9/chars.txt'
|
|
JSON_FILE: {'train': '../textGenerator/dataset/dataset9/train.txt', 'val': '../textGenerator/dataset/dataset9/val.txt'}
|
|
# JSON_FILE: {'train': 'H:/DL-DATASET/360M/train.txt', 'val': 'H:/DL-DATASET/360M/test.txt'}
|
|
SCALE_FACTOR: 0.25
|
|
ROT_FACTOR: 30
|
|
STD: 0.193
|
|
MEAN: 0.588
|
|
ALPHABETS: ''
|
|
|
|
TRAIN:
|
|
BATCH_SIZE_PER_GPU: 32
|
|
SHUFFLE: True
|
|
BEGIN_EPOCH: 0
|
|
END_EPOCH: 100
|
|
RESUME:
|
|
IS_RESUME: False
|
|
FILE: 'output/360CC/crnn/2023-04-27-13-01/checkpoints/checkpoint_99_acc_0.5030.pth'
|
|
OPTIMIZER: 'adam'
|
|
LR: 0.0001
|
|
WD: 0.0
|
|
LR_STEP: [60, 80]
|
|
LR_FACTOR: 0.1
|
|
MOMENTUM: 0.0
|
|
NESTEROV: False
|
|
RMSPROP_ALPHA:
|
|
RMSPROP_CENTERED:
|
|
FINETUNE:
|
|
IS_FINETUNE: False
|
|
FINETUNE_CHECKPOINIT: 'output/checkpoints/mixed_second_finetune_acc_97P7.pth'
|
|
FREEZE: true
|
|
|
|
TEST:
|
|
BATCH_SIZE_PER_GPU: 16
|
|
SHUFFLE: True # for random test rather than test on the whole validation set
|
|
NUM_TEST_BATCH: 1000
|
|
NUM_TEST_DISP: 10
|
|
|
|
MODEL:
|
|
NAME: 'crnn'
|
|
IMAGE_SIZE:
|
|
OW: 160 # origial width: 280
|
|
H: 32
|
|
W: 160 # resized width: 160
|
|
NUM_CLASSES: 0
|
|
NUM_HIDDEN: 256
|
|
|