Upload test_text_upscale_vary_dim768_layer3_pd_4khz_nq9/metrics/config.json with huggingface_hub

test_text_upscale_vary_dim768_layer3_pd_4khz_nq9/metrics/config.json

@@ -0,0 +1,37 @@
+{
+    "train_dataset": "./data/test_output.csv",
+    "valid_dataset": "./data/eval_dataset.csv",
+    "train_duration": 3.0,
+    "valid_test_duration": 3.0,
+    "dataset_chunk_size": 4096,
+    "train_loader_length": null,
+    "sample_rate": 44100,
+    "lr_sample_rate": 4000,
+    "batch_size": 64,
+    "num_workers": 16,
+    "num_epochs": 500,
+    "steps_per_sampler": 500000,
+    "learning_rate": 3e-05,
+    "model_sample_size": 280,
+    "in_channels": 2048,
+    "out_channels": 1024,
+    "model_dim": 768,
+    "layers_num": 3,
+    "cross_attention_dim": 768,
+    "nq": null,
+    "adam_beta1": 0.9,
+    "adam_beta2": 0.999,
+    "adam_weight_decay": 0.01,
+    "adam_epsilon": 1e-08,
+    "lr_scheduler_type": "cosine",
+    "num_warmup_steps": 0,
+    "gradient_accumulation_steps": 1,
+    "max_train_steps": 250000000,
+    "num_train_epochs": 500,
+    "device": "cuda",
+    "experiment_dir": "./exp",
+    "experiment_name": "baseline_exp1",
+    "checkpoint_dir": "checkpoints",
+    "test_output_dir": "test_outputs",
+    "codec_weight_path": "/root/.cache/descript/dac/weights_44khz_8kbps_0.0.1.pth"
+}