-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathexp_finetuning.yaml
More file actions
30 lines (30 loc) · 874 Bytes
/
exp_finetuning.yaml
File metadata and controls
30 lines (30 loc) · 874 Bytes
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
path: "results/fine-tuning80x80/"
# +x is distance
# +y is left
# +z is top
targets:
x : [0.0] #, 1.0] #, 1.0] #, 1.0] #, 1.0, 1.0, 1.0, 1.0, 1.0, 2.0]
y : [0.0] #, -1.0] #, 1.0] #, 0.0] #, 0.0, -1.0, -1.0, 1.0, 1.0, 0.0]
z : [0.0] #, 0.0] #, 0.0] #, -1.0] #, 1.0, -1.0, 1.0, -1.0, 1.0, 0.0]
stlc: # small-time locally controllability
offsets:
# base
- target: [0,0,0]
position: [0,0,0]
lr_pos: 0.001
lr_patch: 0.01
prob_weight: 1.0
num_hl_iter: 10
num_pos_restarts: 3
num_pos_epochs: 1
num_patch_epochs: 1
batch_size: 32
num_patches: 1
scale_min: 0.2
scale_max: 0.7
patch:
mode: 'diffusion' # face, white, random, diffusion
size: [80, 80] # patch size in [height, width]
path: 'src/custom_patches/diffusion_patches_100.npy' # if mode == 'face' set a path, otherwise path is ignored
mode: "joint" # fixed, joint, split, hybrid
quantized: false