-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathOURS.sh
83 lines (77 loc) · 1.72 KB
/
OURS.sh
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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
#--------------------------------2S2F--------------------------------
phase=TimeSelection
# phase=LearnDynamics
tau_s=1.0
slow_dim=2
koopman_dim=2
fast=1
model=ours
system=2S2F
trace_num=200
total_t=5.1
dt=0.01
embed_dim=64
alpha=0.1
id_epoch=100
learn_epoch=150
seed_num=10
tau_1=0.1
tau_N=3.0
device=cpu
gpu_id=0
cpu_num=1
data_dir=Data/$system/data/
id_log_dir=logs/$system/TimeSelection/embed_dim_${embed_dim}/
learn_log_dir=logs/$system/LearnDynamics/slow_${slow_dim}_koopman_${koopman_dim}/
result_dir=Results/$system/slow_${slow_dim}_koopman_${koopman_dim}/fast_${fast}/
# #--------------------------------1S2F--------------------------------
# # phase=TimeSelection
# phase=LearnDynamics
# tau_s=3.0
# slow_dim=1
# koopman_dim=1
# fast=1
# model=ours
# system=1S2F
# trace_num=1
# total_t=15.1
# dt=0.01
# embed_dim=64
# alpha=0.1
# id_epoch=30
# learn_epoch=50
# seed_num=10
# tau_1=0.3
# tau_N=7.0
# device=cpu
# gpu_id=1
# cpu_num=1
# data_dir=Data/$system/data/
# id_log_dir=logs/$system/TimeSelection/embed_dim_${embed_dim}/
# learn_log_dir=logs/$system/LearnDynamics/slow_${slow_dim}_koopman_${koopman_dim}/
# result_dir=Results/$system/slow_${slow_dim}_koopman_${koopman_dim}/fast_${fast}/
CUDA_VISIBLE_DEVICES=$gpu_id python run.py \
--model $model \
--fast $fast \
--system $system \
--trace_num $trace_num \
--total_t $total_t \
--dt $dt \
--embed_dim $embed_dim \
--alpha $alpha \
--id_epoch $id_epoch \
--learn_epoch $learn_epoch \
--seed_num $seed_num \
--tau_1 $tau_1 \
--tau_N $tau_N \
--tau_s $tau_s \
--slow_dim $slow_dim \
--koopman_dim $koopman_dim \
--device $device \
--phase $phase \
--cpu_num $cpu_num \
--data_dir $data_dir \
--id_log_dir $id_log_dir \
--learn_log_dir $learn_log_dir \
--result_dir $result_dir \
# --parallel