Selaa lähdekoodia

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

Dimitri Korsch 5 vuotta sitten
vanhempi
commit
598aec0682
1 muutettua tiedostoa jossa 1 lisäystä ja 1 poistoa
  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
 		)
 
-		if opts.decay:
+		if opts.decay > 0:
 			reg_kwargs = {}
 			if opts.l1_loss:
 				reg_cls = Lasso