Skip to content

GitLab

Unverified Commit a44a1630 authored by Francisco Massa's avatar Francisco Massa Committed by GitHub
Browse files

Fix lint for C++ (#1283)

parent 5537790b
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
torchvision/csrc/models/densenet.cpp
View file @ a44a1630
...@@ -76,8 +76,9 @@ struct _TransitionImpl : torch::nn::SequentialImpl { ...@@ -76,8 +76,9 @@ struct _TransitionImpl : torch::nn::SequentialImpl {
torch::nn::Conv2d(Options(num_input_features, num_output_features, 1) torch::nn::Conv2d(Options(num_input_features, num_output_features, 1)
.stride(1) .stride(1)
.with_bias(false))); .with_bias(false)));
push_back( push_back("pool", torch::nn::Functional([](torch::Tensor input) {
"pool", torch::nn::Functional([](torch::Tensor input) { return torch::avg_pool2d(input, 2, 2, 0, false, true); })); return torch::avg_pool2d(input, 2, 2, 0, false, true);
}));
} }
torch::Tensor forward(torch::Tensor x) { torch::Tensor forward(torch::Tensor x) {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment