-
Notifications
You must be signed in to change notification settings - Fork 0
/
quadconv_config.yaml
36 lines (32 loc) · 1.02 KB
/
quadconv_config.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
#Experiment configuration file.
_spatial_dim: &spatial_dim 3
_num_points: &num_points 44121
_out0: &out0 19456
_out1: &out1 2432
_out2: &out2 304
_latent_dim: &latent_dim 1400
_ratio: &ratio [1,19,2]
#Model arguments
module: "pool"
spatial_dim: *spatial_dim #int
latent_dim: *latent_dim #int
point_seq: [*num_points, *out0, *out1] #[int]
quad_map: "param_quad" #str <function in quadrature.py>
quad_args:
ratio: *ratio
stages: 2 #int
loss_fn: "MSELoss" #str, loss from torch.nn
optimizer: "Adam" #str, optimizer from torch.optim
learning_rate: 0.01 #float
output_activation: "Identity" # str, activation from torch.nn (Identity, Tanh)
conv_params:
in_points: [*num_points, *out0, *out1] #list[int]
out_points: [*out0, *out1, *out2] #list[int]
in_channels: [4, 16, 16] #list[int]
out_channels: [16, 16, 16] #list[int]
bias: [True] #[bool]
filter_seq: [[16, 32, 32, 16]] #[[int]]
filter_mode: [single] #[string] <single|share_in|nested>
decay_param: [1.e+7, 5.e+7, 5.e+7, 5.e+6, 5.e+6] #[float]
block_args:
ratio: *ratio