|
|
|
@ -15,8 +15,10 @@ limitations under the License. */
|
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
|
#include <string>
|
|
|
|
|
#include <vector>
|
|
|
|
|
#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
|
|
|
|
|