Simplify Machine Translation demo by using Trainer API

7 years ago · b0868af5d7
parent 868bdc9771
commit b0868af5d7
3 changed files with 327 additions and 0 deletions
--- a/python/paddle/fluid/tests/book/high-level-api/CMakeLists.txt
+++ b/python/paddle/fluid/tests/book/high-level-api/CMakeLists.txt
@ -8,3 +8,4 @@ endforeach()

 add_subdirectory(fit_a_line)
 add_subdirectory(recognize_digits)
+add_subdirectory(machine_translation)
--- a/python/paddle/fluid/tests/book/high-level-api/machine_translation/CMakeLists.txt
+++ b/python/paddle/fluid/tests/book/high-level-api/machine_translation/CMakeLists.txt
@ -0,0 +1,7 @@
+file(GLOB TEST_OPS RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}" "test_*.py")
+string(REPLACE ".py" "" TEST_OPS "${TEST_OPS}")
+
+# default test
+foreach(src ${TEST_OPS})
+    py_test(${src} SRCS ${src}.py)
+endforeach()
--- a/python/paddle/fluid/tests/book/high-level-api/machine_translation/test_machine_translation.py
+++ b/python/paddle/fluid/tests/book/high-level-api/machine_translation/test_machine_translation.py