removed verbose output

xiph · Jan 5, 2024 · 2d80d1c · 2d80d1c
1 parent 11128e7
commit 2d80d1c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dnn/torch/osce/engine/engine.py b/dnn/torch/osce/engine/engine.py
@@ -48,7 +48,7 @@ def train_one_epoch(model, criterion, optimizer, dataloader, device, scheduler,
 
             # sparsification
             if hasattr(model, 'sparsify'):
-                model.sparsify(verbose=True)
+                model.sparsify()
 
             # update running loss
             running_loss += float(loss.cpu())