-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathparams.yaml
42 lines (36 loc) · 1.1 KB
/
params.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
30
31
32
33
34
35
36
37
38
39
40
41
42
random_seed: 4732
gengaus:
batch_size: 100
save_batch_size: None # should be divisible by batch_size;
# dataset.n_samples by default, don't change it. It is n't supported
env:
pressure_profile:
n_gauses: 10
size_kof: 5
sen_geometry:
distance: 0.5 # between fibers in mm
x_len: 64 # in pixels
y_len: 64 # in pixels
n_angles: 4
n_spl: 1 # number of pictures to show in testing mode
phys:
fiber_sensibility:
value: 0.6 # width of fiber in mm
accuracy: 5 # it is (size of kernel)/value
kof: 1 # between apply force and loss in grams
relative_noise: 0.00 # in fiber transmission detection
noise: 0.00 # in fiber transmission detection
dataset:
n_samples: 1_000
sim:
random_rot: 1
batch_size: 100 # if None - gengaus.batchsize/n_angles/2
test_size: None
# max_possible_size: 20000
test_mod: false # verbose mod for testing sim
train:
batch_size: 7
learning_rate: 0.0001
n_epochs: 2
model_name: TorchSensorNN5S_norm_deep
test_size: None # if None, 1/20 fraction of dataset