Update app.py
Browse files
app.py
CHANGED
|
@@ -62,9 +62,7 @@ from keras import layers
|
|
| 62 |
from keras.applications import MobileNetV2
|
| 63 |
from keras.layers import TextVectorization
|
| 64 |
|
| 65 |
-
|
| 66 |
keras.utils.set_random_seed(111)
|
| 67 |
-
from tensorflow.keras.optimizers.schedules import LearningRateSchedule
|
| 68 |
|
| 69 |
from gtts import gTTS
|
| 70 |
|
|
@@ -579,7 +577,9 @@ early_stopping = keras.callbacks.EarlyStopping(patience=3, restore_best_weights=
|
|
| 579 |
|
| 580 |
|
| 581 |
# Programador de tasa de aprendizaje para el optimizador
|
| 582 |
-
|
|
|
|
|
|
|
| 583 |
def __init__(self, post_warmup_learning_rate, warmup_steps):
|
| 584 |
super().__init__()
|
| 585 |
self.post_warmup_learning_rate = post_warmup_learning_rate
|
|
|
|
| 62 |
from keras.applications import MobileNetV2
|
| 63 |
from keras.layers import TextVectorization
|
| 64 |
|
|
|
|
| 65 |
keras.utils.set_random_seed(111)
|
|
|
|
| 66 |
|
| 67 |
from gtts import gTTS
|
| 68 |
|
|
|
|
| 577 |
|
| 578 |
|
| 579 |
# Programador de tasa de aprendizaje para el optimizador
|
| 580 |
+
from tensorflow.keras.optimizers.schedules import LearningRateSchedule
|
| 581 |
+
|
| 582 |
+
class LRSchedule(LearningRateSchedule):
|
| 583 |
def __init__(self, post_warmup_learning_rate, warmup_steps):
|
| 584 |
super().__init__()
|
| 585 |
self.post_warmup_learning_rate = post_warmup_learning_rate
|