- minor fix "NES.utils.BestWeights"

sgrubas · Dec 30, 2023 · 76c0bfe · 76c0bfe
1 parent 88073af
commit 76c0bfe
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/NES/utils.py b/NES/utils.py
@@ -302,18 +302,15 @@ class BestWeights(tf.keras.callbacks.Callback):
     """
 
     def __init__(self,
-                 nn_model,
                  monitor='loss',
                  freq=50,
                  verbose=1,
                  conversion=lambda x: x * 10**(-0.16),
-
                  ):
         super(BestWeights, self).__init__()
         assert monitor == 'loss' or monitor == 'val_loss', \
             "Only 'loss' and 'val_loss' are supported for monitor metric"
 
-        self.model = nn_model
         self.monitor = monitor
         self.verbose = verbose
         self.freq = freq