Преглед на файлове

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

Dimitri Korsch преди 5 години
родител
ревизия
598aec0682
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  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