From 37f196a7858631ca8b9e22d062470f064d6d64d5 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Tue, 29 Nov 2022 15:40:35 +0000 Subject: [PATCH] LossCrossentropy: add kwargs --- aimodel/src/lib/ai/components/LossCrossentropy.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/aimodel/src/lib/ai/components/LossCrossentropy.py b/aimodel/src/lib/ai/components/LossCrossentropy.py index 359e86f..957454e 100644 --- a/aimodel/src/lib/ai/components/LossCrossentropy.py +++ b/aimodel/src/lib/ai/components/LossCrossentropy.py @@ -9,8 +9,8 @@ class LossCrossentropy(tf.keras.losses.Loss): Args: batch_size (integer): The batch size (currently unused). """ - def __init__(self, batch_size): - super(LossCrossentropy, self).__init__() + def __init__(self, batch_size, **kwargs): + super(LossCrossentropy, self).__init__(**kwargs) self.param_batch_size = batch_size