youngdicey commited on
Commit
aea7245
1 Parent(s): 9d55ab9

Upload unet_sr_baseline/metrics/config.json with huggingface_hub

Browse files
unet_sr_baseline/metrics/config.json ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "train_dataset": "./data/train_dataset.csv",
3
+ "valid_dataset": "./data/eval_dataset.csv",
4
+ "train_duration": 0.38,
5
+ "valid_test_duration": 3.0,
6
+ "train_loader_length": null,
7
+ "batch_size": 32,
8
+ "num_workers": 16,
9
+ "num_epochs": 500,
10
+ "steps_per_sampler": 50000,
11
+ "learning_rate": 0.0001,
12
+ "encoder_dim": 64,
13
+ "encoder_rates": [
14
+ 2,
15
+ 4,
16
+ 8,
17
+ 8
18
+ ],
19
+ "latent_dim": null,
20
+ "decoder_dim": 512,
21
+ "decoder_rates": [
22
+ 8,
23
+ 8,
24
+ 4,
25
+ 2
26
+ ],
27
+ "n_codebooks": 4,
28
+ "codebook_size": 1024,
29
+ "codebook_dim": 8,
30
+ "quantizer_dropout": 0.5,
31
+ "sample_rate": 48000,
32
+ "commitment_loss_weight": 0.25,
33
+ "codebook_loss_weight": 1.0,
34
+ "l1loss": 1.0,
35
+ "stftloss": 0.0,
36
+ "melloss": 1.0,
37
+ "adam_beta1": 0.9,
38
+ "adam_beta2": 0.999,
39
+ "adam_weight_decay": 0.01,
40
+ "adam_epsilon": 1e-08,
41
+ "lr_scheduler_type": "cosine",
42
+ "num_warmup_steps": 0,
43
+ "gradient_accumulation_steps": 1,
44
+ "max_train_steps": 25000000,
45
+ "num_train_epochs": 500,
46
+ "device": "cuda",
47
+ "experiment_dir": "./exp",
48
+ "experiment_name": "mel1_wave1_no_adv",
49
+ "checkpoint_dir": "checkpoints",
50
+ "test_output_dir": "test_outputs"
51
+ }