Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add Example Notebook for Training Custom Watermarking Model on Public Dataset #59

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

hastagAB
Copy link
Contributor

@hastagAB hastagAB commented Dec 1, 2024

Add a new example notebook that provides a comprehensive, step-by-step guide to training a watermarking model using AudioCraft and Dora. The example uses the Librispeech dataset and covers all essential steps, including:

  • 🚀 Setting up the environment and installing dependencies (Torch, Torchaudio, Hydra, Dora, AudioCraft, etc.)
  • 📂 Preparing the Librispeech dataset for training, including manifest creation in AudioCraft format.
  • 📄 Creating YAML configurations for dataset and training management.
  • 🏋️ Running a single-epoch training job using Dora to demonstrate the training process.
  • 📊 Evaluating the trained model checkpoint for performance assessment.

This notebook aims to help users get started with training and evaluating watermarking models effectively, using the robust capabilities provided by AudioCraft and Dora.

🚫 The notebook was prepared on Google Colab which has resources limitations. his code is designed to run effectively on systems with higher computational resources.

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Meta Open Source bot. label Dec 1, 2024
@hastagAB hastagAB force-pushed the feat/customTraining branch from 4f8c4aa to 9e3fa8d Compare December 1, 2024 17:12
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed This label is managed by the Meta Open Source bot.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants