AhmadT198 commited on
Commit
82d14d8
·
verified ·
1 Parent(s): 435ae7b

Adjust GPU duration and number of epochs

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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=200)
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=20,
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
  ]