young-flame-boi commited on
Commit
8e53b47
·
verified ·
1 Parent(s): 4e2ed55

Add LA-2A S6-16d model

Browse files
la2a/s6_16d_release/lightning_logs/version_0/checkpoints/epoch=149-step=442200.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:294d459ec688c9b4361c0c97a05f530d8c4574187e29c429f10d4dfa247a6dc2
3
+ size 61814
la2a/s6_16d_release/lightning_logs/version_0/events.out.tfevents.1761350757.yf.910102.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8a9acce8867837707ba0d93bf1d5be7dd68d49ad9a8cd8a9d77ab85609bc401a
3
+ size 476601237
la2a/s6_16d_release/lightning_logs/version_0/hparams.yaml ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ lr: 0.0003
2
+ train_loss: l1+stft
3
+ save_dir: null
4
+ num_examples: 4
5
+ scheduler: CosineAnnealingLR
6
+ nparams: 2
7
+ ninputs: 1
8
+ noutputs: 1
9
+ window_size: 64
10
+ nfft: 128
11
+ d_model: 2
12
+ mamba_expand: 2
13
+ mamba_dconv: 4
14
+ mamba_states: 16
15
+ stateful: true
16
+ max_streams: 32
17
+ exp_name: s6_16d_release
18
+ model_type: mamba_raw
19
+ d_state: 16
20
+ causal: false
21
+ max_epochs: 150
22
+ preload: true
23
+ sample_rate: 44100
24
+ shuffle: true
25
+ train_subset: train
26
+ val_subset: val
27
+ eval_subset: test
28
+ train_length: 65536
29
+ train_fraction: 1.0
30
+ eval_length: 131072
31
+ batch_size: 16
32
+ num_workers: 0
33
+ precision: 32
34
+ eval_precision: 16
35
+ eval_audo_save_dir: ./audio