Move quantizer defaults to model factory

Author: lgeiger

larq_zoo/core/model_factory.py

@@ -12,6 +12,10 @@
 class ModelFactory:
     """A base class for Larq Zoo models. Defines some common fields."""
 
+    input_quantizer = None
+    kernel_quantizer = None
+    kernel_constraint = None
+
     # This field is included for automatic inference of `num_clases`, if no
     # value is otherwise provided. We set `allow_missing` because we don't want
     # to throw an error if a dataset is not provided, as long as `num_classes`

larq_zoo/training/knowledge_distillation/knowledge_distillation.py

@@ -296,10 +296,6 @@ class TeacherStudentModelFactory(ModelFactory):
     teacher_model: tf.keras.models.Model = ComponentField(allow_missing=True)
     student_model: tf.keras.models.Model = ComponentField()
 
-    input_quantizer = None
-    kernel_quantizer = None
-    kernel_constraint = None
-
     # Must be set if there is a teacher and allow_missing teacher weights is not True.
     # Either a full path or the name of a network (in which case it will be sought in the current `model_dir`).
     initialize_teacher_weights_from: str = Field(allow_missing=True)