Skip to content

Commit 8e1de9c

Browse files
committed
Fix kernel regularizers
1 parent 5d4e332 commit 8e1de9c

File tree

2 files changed

+2
-6
lines changed

2 files changed

+2
-6
lines changed

larq_zoo/literature/real_to_bin_nets.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ class _SharedBaseFactory(ModelFactory, metaclass=ABCMeta):
2121
model_name: str = Field()
2222
momentum: float = Field(0.99)
2323
kernel_initializer: str = Field("glorot_normal")
24-
kernel_regularizer: Optional[tf.keras.regularizers.Regularizer] = Field(None)
24+
kernel_regularizer = None
2525

2626
def first_block(
2727
self, x: tf.Tensor, use_prelu: bool = True, name: str = ""

larq_zoo/literature/xnornet.py

+1-5
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
import larq as lq
44
import tensorflow as tf
5-
from zookeeper import Field, factory
5+
from zookeeper import factory
66

77
from larq_zoo.core import utils
88
from larq_zoo.core.model_factory import ModelFactory
@@ -28,10 +28,6 @@ class XNORNetFactory(ModelFactory):
2828
kernel_quantizer = "xnor_weight_scale"
2929
kernel_constraint = "weight_clip"
3030

31-
kernel_regularizer: Optional[tf.keras.regularizers.Regularizer] = Field(
32-
lambda: tf.keras.regularizers.l2(5e-7)
33-
)
34-
3531
@property
3632
def kernel_regularizer(self):
3733
return tf.keras.regularizers.l2(5e-7)

0 commit comments

Comments
 (0)