index d192d75093d5bf39f71b26dff88883f8bd588842..da5632dece6a4de0147071ad7ba7c0f38bc012b3 100644 (file)
layer_param.set_kernelsize(3);
layer_param.set_stride(2);
Im2colLayer<TypeParam> layer(layer_param);
+ Caffeine::set_mode(Caffeine::CPU);
layer.SetUp(this->blob_bottom_vec_, &(this->blob_top_vec_));
layer.Forward(this->blob_bottom_vec_, &(this->blob_top_vec_));
// We are lazy and will only check the top left block
LayerParameter layer_param;
layer_param.set_kernelsize(3);
layer_param.set_stride(2);
+ Caffeine::set_mode(Caffeine::CPU);
Im2colLayer<TypeParam> layer(layer_param);
GradientChecker<TypeParam> checker(1e-2, 1e-2);
checker.CheckGradient(layer, this->blob_bottom_vec_, this->blob_top_vec_);