| 18 | } |
| 19 | |
| 20 | inline torch::nn::UpsampleOptions upsample_options(std::vector<double> scale_size,bool align_corners = true){ |
| 21 | torch::nn::UpsampleOptions upsample_options = torch::nn::UpsampleOptions(); |
| 22 | upsample_options.scale_factor(scale_size); |
| 23 | upsample_options.mode(torch::kBilinear).align_corners(align_corners); |
| 24 | return upsample_options; |
| 25 | } |
| 26 | |
| 27 | inline torch::nn::Dropout2dOptions dropout_options(float p, bool inplace){ |
| 28 | torch::nn::Dropout2dOptions dropoutoptions(p); |
no outgoing calls
no test coverage detected