youngdicey commited on
Commit
bc29c0c
·
verified ·
1 Parent(s): 71e09f9

Upload test_upscale_vary/metrics/config.json with huggingface_hub

Browse files
test_upscale_vary/metrics/config.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ "batch_size": 64,
10
+ "num_workers": 16,
11
+ "num_epochs": 500,
12
+ "steps_per_sampler": 50000,
13
+ "learning_rate": 0.0001,
14
+ "model_sample_size": 280,
15
+ "in_channels": 2048,
16
+ "out_channels": 1024,
17
+ "adam_beta1": 0.9,
18
+ "adam_beta2": 0.999,
19
+ "adam_weight_decay": 0.01,
20
+ "adam_epsilon": 1e-08,
21
+ "lr_scheduler_type": "cosine",
22
+ "num_warmup_steps": 0,
23
+ "gradient_accumulation_steps": 1,
24
+ "max_train_steps": 25000000,
25
+ "num_train_epochs": 500,
26
+ "device": "cuda",
27
+ "experiment_dir": "./exp",
28
+ "experiment_name": "baseline_exp1",
29
+ "checkpoint_dir": "checkpoints",
30
+ "test_output_dir": "test_outputs",
31
+ "codec_weight_path": "/root/.cache/descript/dac/weights_44khz_8kbps_0.0.1.pth"
32
+ }