youngdicey commited on
Commit
c868eee
1 Parent(s): ba3665a

Upload test_text_upscale_vary_dim768_layer3_pd_16khz_nq9/metrics/config.json with huggingface_hub

Browse files
test_text_upscale_vary_dim768_layer3_pd_16khz_nq9/metrics/config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "train_dataset": "./data/train_dataset.csv",
3
+ "valid_dataset": "./data/eval_dataset.csv",
4
+ "train_duration": 3.0,
5
+ "valid_test_duration": 3.0,
6
+ "dataset_chunk_size": 4096,
7
+ "train_loader_length": null,
8
+ "sample_rate": 44100,
9
+ "lr_sample_rate": 4000,
10
+ "batch_size": 64,
11
+ "num_workers": 16,
12
+ "num_epochs": 500,
13
+ "steps_per_sampler": 500000,
14
+ "learning_rate": 3e-05,
15
+ "model_sample_size": 280,
16
+ "in_channels": 2048,
17
+ "out_channels": 1024,
18
+ "model_dim": 768,
19
+ "layers_num": 6,
20
+ "nq": null,
21
+ "adam_beta1": 0.9,
22
+ "adam_beta2": 0.999,
23
+ "adam_weight_decay": 0.01,
24
+ "adam_epsilon": 1e-08,
25
+ "lr_scheduler_type": "cosine",
26
+ "num_warmup_steps": 0,
27
+ "gradient_accumulation_steps": 1,
28
+ "max_train_steps": 250000000,
29
+ "num_train_epochs": 500,
30
+ "device": "cuda",
31
+ "experiment_dir": "./exp",
32
+ "experiment_name": "baseline_exp1",
33
+ "checkpoint_dir": "checkpoints",
34
+ "test_output_dir": "test_outputs",
35
+ "codec_weight_path": "/root/.cache/descript/dac/weights_44khz_8kbps_0.0.1.pth"
36
+ }