-
Notifications
You must be signed in to change notification settings - Fork 5
/
Copy pathHrnet_Ocr_aug_pseudo.json
42 lines (42 loc) · 1.91 KB
/
Hrnet_Ocr_aug_pseudo.json
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
{
"path_dataset_root": "/opt/ml/segmentation/input/data/",
"path_json_train": "/opt/ml/segmentation/input/data/train_pseudo.json",
"path_json_valid": "/opt/ml/segmentation/input/data/val.json",
"path_json_test": "/opt/ml/segmentation/input/data/test.json",
"path_project_root": "./Projects/Hrnet_Ocr_train_Aug_pseudo",
"random_fix": true,
"random_seed": 21,
"model": "Hrnet_Seg_Ocr",
"train_model_pretrained": true,
"train_model_pretrained_path": "./Projects/Hrnet_Ocr_train_Aug/best_score.pt",
"train_model_num_epochs": 50,
"train_model_batch_size": 8,
"train_loss": "CrossEntropyLoss",
"train_optimizer": "Adam",
"train_optimizer_learning_rate": 0.0001,
"train_optimizer_weight_decay": 1e-06,
"train_data_num_workers": 2,
"train_data_loading_mode": "preload",
"train_data_transform_preprocess_train": "HorizontalFlip_Rotate90",
"train_data_transform_preprocess_train_object_aug": false,
"train_data_transform_preprocess_valid": "Default",
"test_path_checkpoint": "./Projects/Hrnet_Ocr_train_Aug_pseudo/best_score.pt",
"test_path_submission": "./Projects/Hrnet_Ocr_train_Aug_pseudo/submission_pseudo.csv",
"test_model_batch_size": 30,
"test_data_num_workers": 4,
"test_data_loading_mode": "realtime",
"test_data_target_size": 256,
"test_data_transform_preprocess": "Default",
"test_data_transform_tta": "HorizontalFlip_Rotate90",
"test_data_transform_dcrf": true,
"test_data_transform_dcrf_num_workers": 16,
"test_data_transform_dcrf_iter": 10,
"test_data_transform_dcrf_gau_sxy": 3,
"test_data_transform_dcrf_gau_compat": 3,
"test_data_transform_dcrf_bi_sxy": 49,
"test_data_transform_dcrf_bi_srgb": 5,
"test_data_transform_dcrf_bi_compat": 4,
"test_pseudo_labeling": false,
"test_pseudo_labeling_threshold_area": 3500,
"test_pseudo_labeling_output_path": "../input/data/train_pseudo.json"
}