diff --git a/paddle/gserver/tests/test_PriorBox.cpp b/paddle/gserver/tests/test_PriorBox.cpp index 8aabb1ef97a21322cb5e42d6efb0d655877c3967..d37c0bb70209a6cf2cf50e071a544f5233f3f8c5 100644 --- a/paddle/gserver/tests/test_PriorBox.cpp +++ b/paddle/gserver/tests/test_PriorBox.cpp @@ -15,8 +15,10 @@ limitations under the License. */ #include #include #include +#include "./paddle/utils/CommandLineParser.h" #include "ModelConfig.pb.h" #include "paddle/gserver/layers/DataLayer.h" +#include "paddle/gserver/layers/ExpandConvTransLayer.h" #include "paddle/math/MathUtils.h" #include "paddle/trainer/Trainer.h" #include "paddle/utils/GlobalConstants.h" @@ -29,7 +31,9 @@ using namespace std; // NOLINT P_DECLARE_bool(use_gpu); P_DECLARE_int32(gpu_id); +P_DECLARE_double(checkgrad_eps); P_DECLARE_bool(thread_local_rand_use_global_seed); +P_DECLARE_bool(prev_batch_state); // Do one forward pass of priorBox layer and check to see if its output // matches the given result