Skip to content

GitLab

"src/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "1e152030bd836e0b2bfc5fabe2186325b8d38a04"
Unverified Commit a44a1630 authored by Francisco Massa's avatar Francisco Massa Committed by GitHub
Browse files

Fix lint for C++ (#1283)

parent 5537790b
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 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) {
......
torchvision/csrc/vision.h
View file @ a44a1630
...@@ -3,4 +3,4 @@ ...@@ -3,4 +3,4 @@
#include <torchvision/models/models.h> #include <torchvision/models/models.h>
#endif // VISION_H #endif // VISION_H
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