Gloria Dal Santo commited on
Commit
7a89c8d
·
1 Parent(s): 56e1924

Update sparsity weight

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -73,7 +73,7 @@ def process_fdn(N, delay_lengths):
73
  log=False
74
  )
75
  trainer.register_criterion(mse_loss(nfft=config.nfft, device=config.device), 1)
76
- trainer.register_criterion(sparsity_loss(), 0.2, requires_model=True)
77
 
78
  ## ---------------- TRAIN ---------------- ##
79
 
 
73
  log=False
74
  )
75
  trainer.register_criterion(mse_loss(nfft=config.nfft, device=config.device), 1)
76
+ trainer.register_criterion(sparsity_loss(), 1, requires_model=True)
77
 
78
  ## ---------------- TRAIN ---------------- ##
79