Caffe::set_mode(Caffe::CPU);
SplitLayer<TypeParam> layer(layer_param);
GradientChecker<TypeParam> checker(1e-2, 1e-2);
- checker.CheckGradientExhaustive(layer, this->blob_bottom_vec_, this->blob_top_vec_);
checker.CheckGradientExhaustive(layer, this->blob_bottom_vec_,
this->blob_top_vec_);
}
Caffe::set_mode(Caffe::GPU);
SplitLayer<TypeParam> layer(layer_param);
GradientChecker<TypeParam> checker(1e-2, 1e-2);
- checker.CheckGradientExhaustive(layer, this->blob_bottom_vec_, this->blob_top_vec_);
checker.CheckGradientExhaustive(layer, this->blob_bottom_vec_,
this->blob_top_vec_);
}
SplitLayer<TypeParam> layer(layer_param);
GradientChecker<TypeParam> checker(1e-2, 1e-2);
this->blob_top_vec_[0] = this->blob_bottom_vec_[0];
- checker.CheckGradientExhaustive(layer, this->blob_bottom_vec_, this->blob_top_vec_);
checker.CheckGradientExhaustive(layer, this->blob_bottom_vec_,
this->blob_top_vec_);
}
SplitLayer<TypeParam> layer(layer_param);
GradientChecker<TypeParam> checker(1e-2, 1e-2);
this->blob_top_vec_[0] = this->blob_bottom_vec_[0];
- checker.CheckGradientExhaustive(layer, this->blob_bottom_vec_, this->blob_top_vec_);
checker.CheckGradientExhaustive(layer, this->blob_bottom_vec_,
this->blob_top_vec_);
}