mattricesound commited on
Commit
53f9ab9
·
1 Parent(s): 2b0b0f8

Fix umx_compressor path. Only log max 10 audio items2

Browse files
Files changed (2) hide show
  1. cfg/exp/umx_compressor +0 -4
  2. 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(