-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconfig.yaml
35 lines (35 loc) · 878 Bytes
/
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
global:
root_dir: ""
iteration: "intent_indoor_baseline_2"
description: "intent_indoor_baseline full run changed run params"
enable_preprocessing: False
seed: 8
enable_intent: True
enable_wandb: True
transformer:
fps: 2
path: "data/"
data_save_file: "data/train" #file path without .npz
enable_benchmark_test: False
test_path: "data/"
test_save_file: "data/train" #file path without .npz
data:
HEIGHT: 64
WIDTH: 64
CHANNELS: 1
SEQUENCE_LENGTH: 10
VAL_INTENT: ""
BENCHMARK_TEST_INTENT: ""
trainer:
BATCH: 64
lr: 0.0001
epochs: 25
lambda: 0.01 #L2 regularization #changed from 1e-4
num_classes: 3
model_save_path: "models/"
model:
name: "ConvLSTM"
convlstm_hidden: "256"
num_conv_lstm_layers: 1
pretrained_path: ""
optimizer_path: ""