File tree Expand file tree Collapse file tree 1 file changed +6
-5
lines changed Expand file tree Collapse file tree 1 file changed +6
-5
lines changed Original file line number Diff line number Diff line change @@ -76,11 +76,12 @@ class InterpolateOpMaker : public framework::OpProtoAndCheckerMaker {
76
76
77
77
AddAttr<int >(" out_h" , " output height of interpolate op." );
78
78
AddAttr<int >(" out_w" , " output width of interpolate op." );
79
- AddAttr<std::string>(
80
- " interp_method" ,
81
- " (string), interpolation method, can be \" bilinear\" for "
82
- " bilinear interpolation and \" nearest\" for nearest "
83
- " neighbor interpolation." );
79
+ AddAttr<std::string>(" interp_method" ,
80
+ " (string, default \" bilinear\" ), interpolation "
81
+ " method, can be \" bilinear\" for "
82
+ " bilinear interpolation and \" nearest\" for nearest "
83
+ " neighbor interpolation." )
84
+ .SetDefault (" bilinear" );
84
85
AddComment (R"DOC(
85
86
This operator samples input X to given output shape by using specified
86
87
interpolation method, the interpolation methods can be \"nearest\"
You can’t perform that action at this time.
0 commit comments