Adjust GPU duration and number of epochs
Browse files
app.py
CHANGED
@@ -48,7 +48,7 @@ index_to_class = dict((v,k) for k,v in class_to_index.items())
|
|
48 |
names_to_ids = lambda labels: np.array([class_to_index.get(x) for x in labels])
|
49 |
train_labels = names_to_ids(labels)
|
50 |
|
51 |
-
@spaces.GPU(duration=
|
52 |
def load_model():
|
53 |
|
54 |
|
@@ -70,7 +70,7 @@ def load_model():
|
|
70 |
val_labels = names_to_ids(val_labels)
|
71 |
h = model.fit(
|
72 |
padded_train_seq, train_labels, validation_data=(val_seq, val_labels),
|
73 |
-
epochs=
|
74 |
callbacks=[
|
75 |
tf.keras.callbacks.EarlyStopping(monitor='val_accuracy', patience=2)
|
76 |
]
|
|
|
48 |
names_to_ids = lambda labels: np.array([class_to_index.get(x) for x in labels])
|
49 |
train_labels = names_to_ids(labels)
|
50 |
|
51 |
+
@spaces.GPU(duration=150)
|
52 |
def load_model():
|
53 |
|
54 |
|
|
|
70 |
val_labels = names_to_ids(val_labels)
|
71 |
h = model.fit(
|
72 |
padded_train_seq, train_labels, validation_data=(val_seq, val_labels),
|
73 |
+
epochs=8,
|
74 |
callbacks=[
|
75 |
tf.keras.callbacks.EarlyStopping(monitor='val_accuracy', patience=2)
|
76 |
]
|