Upload 2 files
Browse files
mvae/mvae_24_split/checkpoint_50000.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f50484147964125fce802468701bc95bcc2c44037f5e1a38a0544576dea6bd99
|
| 3 |
+
size 26486009
|
mvae/mvae_24_split/config.yaml
ADDED
|
@@ -0,0 +1,22 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
h5_path: /home/sizhe/actions.h5
|
| 2 |
+
window_size: 10
|
| 3 |
+
num_workers: 4
|
| 4 |
+
train_split_ratio: 0.8
|
| 5 |
+
batch_size: 64
|
| 6 |
+
learning_rate: 0.0001
|
| 7 |
+
grad_clip: 1.0
|
| 8 |
+
ema_decay: 0.999
|
| 9 |
+
use_amp: false
|
| 10 |
+
total_steps: 50000
|
| 11 |
+
weight_kl: 1.0e-06
|
| 12 |
+
log_interval: 100
|
| 13 |
+
val_interval: 10000
|
| 14 |
+
save_interval: 10000
|
| 15 |
+
resume_checkpoint: null
|
| 16 |
+
exp_name: mvae_24
|
| 17 |
+
save_dir: ./mvae
|
| 18 |
+
seed: 0
|
| 19 |
+
device: cuda
|
| 20 |
+
num_layers: 7
|
| 21 |
+
latent_tokens: 1
|
| 22 |
+
latent_dim: 24
|