Update sample.json

abijaz · Aug 16, 2024 · 0cf470c · 0cf470c
1 parent de74fca
commit 0cf470c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...Analytics/Features-and-Functionality/IntelTensorFlow_TextGeneration_with_LSTM/sample.json b/...Analytics/Features-and-Functionality/IntelTensorFlow_TextGeneration_with_LSTM/sample.json
@@ -23,7 +23,7 @@
             "id": "inc_text_generation_lstm_py",
             "steps": [
                 "export ITEX_ENABLE_NEXTPLUGGABLE_DEVICE=0",
-                "python TextGenerationModelTraining.py"
+                "ITEX_NUM_EPOCHS=5 KERAS_NUM_EPOCHS=5 python TextGenerationModelTraining.py"
              ]
         }
       ]