Demosthene-OR commited on
Commit
2bc4127
1 Parent(s): e753237

Update main_dl.py

Browse files
Files changed (1) hide show
  1. main_dl.py +2 -3
main_dl.py CHANGED
@@ -231,10 +231,9 @@ def load_all_data():
231
  transformer_en_fr.compile(optimizer="rmsprop", loss="sparse_categorical_crossentropy", metrics=["accuracy"])
232
  transformer_fr_en.compile(optimizer="rmsprop", loss="sparse_categorical_crossentropy", metrics=["accuracy"])
233
 
234
- return translation_en_fr, translation_fr_en, rnn_en_fr, rnn_fr_en, transformer_en_fr, transformer_fr_en
235
 
236
- n1 = 0
237
- translation_en_fr, translation_fr_en, rnn_en_fr, rnn_fr_en, transformer_en_fr, transformer_fr_en = load_all_data()
238
 
239
 
240
  def display_translation(n1, Lang,model_type):
 
231
  transformer_en_fr.compile(optimizer="rmsprop", loss="sparse_categorical_crossentropy", metrics=["accuracy"])
232
  transformer_fr_en.compile(optimizer="rmsprop", loss="sparse_categorical_crossentropy", metrics=["accuracy"])
233
 
234
+ return rnn_en_fr, rnn_fr_en, transformer_en_fr, transformer_fr_en
235
 
236
+ rnn_en_fr, rnn_fr_en, transformer_en_fr, transformer_fr_en = load_all_data()
 
237
 
238
 
239
  def display_translation(n1, Lang,model_type):