| Parameter | Description | Value |
|---|---|---|
| Pretrained Model | Name of the pre-trained MusicGen model used for fine-tuning. | facebook/musicgen-medium |
| Dataset Path | Path to the CSV file containing metadata for training. | /home/shivam.chauhan/Music/Atharva/Processed_Dataset/Makam_32KHz/metadata.csv |
| Audio Base Path | Directory containing audio files for training. | /home/shivam.chauhan/Music/Atharva/Processed_Dataset/Makam_32KHz/ |
| Model Save Path | Path where the fine-tuned model will be saved. | ./ModelsFinetuned/MusicgenMedium_with_adapters_EncoderDecoder_newMaqam.pt |
| Adapter Bottleneck Dim | Size of the bottleneck layer in the adapter. | 32 |
| Batch Size | Number of samples per training batch. | 4 |
| Learning Rate | Step size for updating model weights. | 5e-5 |
| Weight Decay | Regularization parameter to prevent overfitting. | 0.05 |
| Number of Epochs | Total number of training iterations over the dataset. | 30 |
| Dropout Probability | Probability of dropping units in adapter layers. | 0.1 |
| Max Gradient Norm | Maximum norm for gradient clipping to prevent explosion. | 1.0 |
| Train-Test Split Ratio | Proportion of data used for training vs validation. | 90:10 |
| Early Stopping Patience | Number of epochs without improvement before stopping training. | 5 epochs |
facebook/musicgen-medium) that is fine-tuned for a specific task.| Parameter | Description | Value |
|---|---|---|
| Pretrained Model | Name of the pre-trained MusicGen model used for inference. | facebook/musicgen-medium |
| Fine-tuned Model Path | Path where the fine-tuned model is stored. | ./ModelsFinetuned/New/MusicgenMedium_with_adapters_EncoderDecoder.pt |
| Output Audio Path | Path where the generated audio file is saved. | ./GeneratedAudios/1.wav |
| Waveform Graph Path | Path where the waveform visualization is stored. | ./GeneratedGraphs/1.jpeg |
| Sample Rate | Desired sample rate for the generated audio. | 16,000 Hz |
| Adapter Bottleneck Dim | Size of the bottleneck layer in the adapter network. | 32 |
| Max New Tokens | Controls the length of the generated music (512 ≈ 10 sec). | 512 |
| Device | Specifies whether to use GPU or CPU for inference. | CUDA if available, else CPU |
| Use Fine-tuned Model | Determines whether to use the fine-tuned model or pre-trained. | True (uses fine-tuned model) |
facebook/musicgen-medium, which is fine-tuned for customized music generation.use_finetuned_model = True, the model loads from this path.pre-trained or fine-tuned based on configuration)..wav file.