Sfoglia il codice sorgente

minor weight decay initialization fix: now WD is disabled if value < 0 given

Dimitri Korsch 5 anni fa
parent
commit
598aec0682
1 ha cambiato i file con 1 aggiunte e 1 eliminazioni
  1. 1 1
      cvfinetune/finetuner/base.py

+ 1 - 1
cvfinetune/finetuner/base.py

@@ -85,7 +85,7 @@ class _ModelMixin(abc.ABC):
 			decay=0, gradient_clipping=False, **opt_kwargs
 			decay=0, gradient_clipping=False, **opt_kwargs
 		)
 		)
 
 
-		if opts.decay:
+		if opts.decay > 0:
 			reg_kwargs = {}
 			reg_kwargs = {}
 			if opts.l1_loss:
 			if opts.l1_loss:
 				reg_cls = Lasso
 				reg_cls = Lasso