Skip to content

Commit

Permalink
Fix reference to the audiocraft grids (#47)
Browse files Browse the repository at this point in the history
Co-authored-by: Tuan Tran <{ID}+{username}@users.noreply.github.com>
  • Loading branch information
antoine-tran and Tuan Tran authored Aug 6, 2024
1 parent c40d504 commit 5fc61ad
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
4 changes: 2 additions & 2 deletions docs/TRAINING.md
Original file line number Diff line number Diff line change
Expand Up @@ -71,7 +71,7 @@ default:
global: your_slurm_partitions
team: your_slurm_partitions
reference_dir: /tmp
darwin: # if we detect we are on a Mac, then most likely we are doing unit testing etc.
darwin: # if we detect we are on a Mac, then most likely we are doing unit testing etc.
dora_dir: [YOUR PATH]
partitions:
global: your_slurm_partitions
Expand Down Expand Up @@ -114,7 +114,7 @@ result, message = detector(watermarked_audio, sr)

## Training the HF AudioSeal model

We also provide the hyperparameter and training config (in Dora term, a "grid") to reproduce our checkpoints for AudioSeal in HuggingFace (which is also the one used to produce the results ported in the ICML paper). To get this, check the AudioCraft's watermarking [grid](https://github.com/facebookresearch/audiocraft/blob/main/audiocraft/grids/watermarking/1315_kbits_seeds.py). To reproduce the result, run the `dora grid` command:
We also provide the hyperparameter and training config (in Dora term, a "grid") to reproduce our checkpoints for AudioSeal in HuggingFace (which is also the one used to produce the results ported in the ICML paper). To get this, check the AudioCraft's watermarking [grid](https://github.com/facebookresearch/audiocraft/blob/main/audiocraft/grids/watermarking/kbits.py). To reproduce the result, run the `dora grid` command:

```bash
AUDIOCRAFT_CONFIG=my_config.yaml AUDIOCRAFT_DSET=audio/voxpopuli dora grid watermarking.1315_kbits_seeds
Expand Down
2 changes: 1 addition & 1 deletion src/audioseal/builder.py
Original file line number Diff line number Diff line change
Expand Up @@ -75,7 +75,7 @@ class AudioSealDetectorConfig:
def as_dict(obj: Any) -> Dict[str, Any]:
if isinstance(obj, dict):
return obj
if is_dataclass(obj):
if is_dataclass(obj) and not isinstance(obj, type):
return asdict(obj)
elif isinstance(obj, DictConfig):
return OmegaConf.to_container(obj) # type: ignore
Expand Down

0 comments on commit 5fc61ad

Please sign in to comment.