Spaces:
Runtime error
Runtime error
Update CALTextModel.py
Browse files- CALTextModel.py +2 -2
CALTextModel.py
CHANGED
@@ -478,9 +478,9 @@ class CALText(layers.Layer):
|
|
478 |
|
479 |
|
480 |
|
481 |
-
class
|
482 |
def __init__(self,training): # Define All your Variables Here. And other configurations
|
483 |
-
super(
|
484 |
self.dense_blocks=3
|
485 |
self.levels_count=16
|
486 |
self.growth=24
|
|
|
478 |
|
479 |
|
480 |
|
481 |
+
class CALText_Model(tf.keras.Model): # Subclass from tf.keras.model
|
482 |
def __init__(self,training): # Define All your Variables Here. And other configurations
|
483 |
+
super(CALText_Model, self).__init__()
|
484 |
self.dense_blocks=3
|
485 |
self.levels_count=16
|
486 |
self.growth=24
|