-
Notifications
You must be signed in to change notification settings - Fork 10
/
mpm_train.yaml
29 lines (23 loc) · 1.04 KB
/
mpm_train.yaml
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
# Hydra configuration file for training
train:
imgs: 'data/train_imgs/' #
mpms: 'data/train_mpms/' #
itvs: [1, 3, 5, 7, 9] # Multiple patterns are OK like [1, 3, 5, 7, 9]
epochs: 300 # Number of epoch
batch_size: 16 # Batch size
lr: 0.0001 # Learning rate of Adam
eval:
imgs: 'data/eval_imgs/' # None is OK. In that case, eval.rate in the training data is used for evaluation.
mpms: 'data/eval_mpms_parallel/' # None is OK
itvs: [9] # Multiple patterns are OK like [1, 3, 5, 7, 9]
rate: 0.2 # Evaluation rate for no eval data
output:
save: True
dir: 'checkpoints/'
load: False # If you use a trained model, please rewrite it.
dataloader:
edge: 10 # Image edge range (This area will not be cropped)
height: 256 # Number of pixel of crop height
width: 256 # Number of pixel of crop width
info:
direction: parallel # Direction type of MPM