forked from branislav1991/PyTorchProjectFramework
-
Notifications
You must be signed in to change notification settings - Fork 0
/
config_segmentation.json
40 lines (40 loc) · 1009 Bytes
/
config_segmentation.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
{
"train_dataset_params": {
"dataset_name": "segmentation2d",
"dataset_path": "",
"loader_params": {
"batch_size": 24,
"shuffle": true,
"num_workers": 4,
"pin_memory": true
},
"input_size": [200, 200]
},
"val_dataset_params": {
"dataset_name": "segmentation2d",
"dataset_path": "",
"loader_params": {
"batch_size": 24,
"shuffle": false,
"num_workers": 4,
"pin_memory": true
},
"input_size": [200, 200]
},
"model_params": {
"model_name": "segmentation2d",
"is_train": true,
"max_epochs": 40,
"lr": 0.01,
"export_path": "",
"checkpoint_path": "",
"load_checkpoint": -1,
"lr_policy": "step",
"lr_decay_iters": 10
},
"visualization_params": {
"name": "2d segmentation"
},
"printout_freq": 10,
"model_update_freq": 1
}