forked from simonalexanderson/StyleGestures
-
Notifications
You must be signed in to change notification settings - Fork 0
/
trinity.yml
48 lines (46 loc) · 918 Bytes
/
trinity.yml
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
43
44
45
46
47
48
seed_everything: null
trainer:
logger: true
default_root_dir: "results/"
max_epochs: 131
max_time: null
limit_train_batches: 1.0
flush_logs_every_n_steps: 100
log_every_n_steps: 50
precision: 32
weights_summary: top
num_sanity_val_steps: 2
model:
hidden_channels: 512
K: 16
actnorm_scale: 1.0
flow_permutation: invconv
flow_coupling: affine
network_model: LSTM
num_layers: 2
LU_decomposed: true
distribution: normal
batch_size: 80
max_grad_clip: 5
max_grad_norm: 100
beta_1: 0.9
beta_2: 0.999
eps: 1.0e-08
warmup_steps: 3000
min_lr: 0.00015
init_lr: 0.0015
checkpoints_gap: 100000000
scalar_log_gaps: 1
save_videos_every_n_epochs: 10
data:
dropout: 0.4
seqlen: 5
n_lookahead: 20
data_root: data/trinity/processed/
x_channels: 45
cond_channels: 927
framerate: 20
num_plot_samples: 5
is_training: true
num_workers: 20
trinity: true