Update custom_objects.py
Browse files- custom_objects.py +1 -1
custom_objects.py
CHANGED
|
@@ -3,7 +3,7 @@ from tensorflow.keras.saving import register_keras_serializable
|
|
| 3 |
from tensorflow.keras import layers, models, backend as K
|
| 4 |
import numpy as np
|
| 5 |
|
| 6 |
-
|
| 7 |
def firetrust_activation(x):
|
| 8 |
return 0.5 + tf.sigmoid(x)
|
| 9 |
|
|
|
|
| 3 |
from tensorflow.keras import layers, models, backend as K
|
| 4 |
import numpy as np
|
| 5 |
|
| 6 |
+
@register_keras_serializable
|
| 7 |
def firetrust_activation(x):
|
| 8 |
return 0.5 + tf.sigmoid(x)
|
| 9 |
|