|
a |
|
b/config.py |
|
|
1 |
conf = { |
|
|
2 |
"WORK_PATH": "./work", |
|
|
3 |
"CUDA_VISIBLE_DEVICES": "0,1,2,3", |
|
|
4 |
"data": { |
|
|
5 |
'dataset_path': "your_dataset_path", |
|
|
6 |
'resolution': '64', |
|
|
7 |
'dataset': 'CASIA-B', |
|
|
8 |
# In CASIA-B, data of subject #5 is incomplete. |
|
|
9 |
# Thus, we ignore it in training. |
|
|
10 |
# For more detail, please refer to |
|
|
11 |
# function: utils.data_loader.load_data |
|
|
12 |
'pid_num': 73, |
|
|
13 |
'pid_shuffle': False, |
|
|
14 |
}, |
|
|
15 |
"model": { |
|
|
16 |
'hidden_dim': 256, |
|
|
17 |
'lr': 1e-4, |
|
|
18 |
'hard_or_full_trip': 'full', |
|
|
19 |
'batch_size': (8, 16), |
|
|
20 |
'restore_iter': 0, |
|
|
21 |
'total_iter': 80000, |
|
|
22 |
'margin': 0.2, |
|
|
23 |
'num_workers': 3, |
|
|
24 |
'frame_num': 30, |
|
|
25 |
'model_name': 'GaitSet', |
|
|
26 |
}, |
|
|
27 |
} |