Spaces:
Sleeping
Sleeping
Commit
·
53f9ab9
1
Parent(s):
2b0b0f8
Fix umx_compressor path. Only log max 10 audio items2
Browse files- cfg/exp/umx_compressor +0 -4
- remfx/models.py +3 -0
cfg/exp/umx_compressor
DELETED
|
@@ -1,4 +0,0 @@
|
|
| 1 |
-
# @package _global_
|
| 2 |
-
defaults:
|
| 3 |
-
- override /model: umx
|
| 4 |
-
- override /effects: compression
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
remfx/models.py
CHANGED
|
@@ -287,10 +287,13 @@ def log_wandb_audio_batch(
|
|
| 287 |
samples: Tensor,
|
| 288 |
sampling_rate: int,
|
| 289 |
caption: str = "",
|
|
|
|
| 290 |
):
|
| 291 |
num_items = samples.shape[0]
|
| 292 |
samples = rearrange(samples, "b c t -> b t c")
|
| 293 |
for idx in range(num_items):
|
|
|
|
|
|
|
| 294 |
logger.experiment.log(
|
| 295 |
{
|
| 296 |
f"{id}_{idx}": wandb.Audio(
|
|
|
|
| 287 |
samples: Tensor,
|
| 288 |
sampling_rate: int,
|
| 289 |
caption: str = "",
|
| 290 |
+
max_items: int = 10,
|
| 291 |
):
|
| 292 |
num_items = samples.shape[0]
|
| 293 |
samples = rearrange(samples, "b c t -> b t c")
|
| 294 |
for idx in range(num_items):
|
| 295 |
+
if idx >= max_items:
|
| 296 |
+
break
|
| 297 |
logger.experiment.log(
|
| 298 |
{
|
| 299 |
f"{id}_{idx}": wandb.Audio(
|