Changed an initialization of avg_pool2d Functional to use lambda (#1083)

487c9bf4 · Igor Fedan · Francisco Massa · 34833427 · 487c9bf4
Commit 487c9bf4 authored Jul 08, 2019 by Igor Fedan Committed by Francisco Massa Jul 08, 2019
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torchvision/csrc/models/densenet.cpp torchvision/csrc/models/densenet.cpp +1 -1

No files found.
--- a/torchvision/csrc/models/densenet.cpp
+++ b/torchvision/csrc/models/densenet.cpp
@@ -77,7 +77,7 @@ struct _TransitionImpl : torch::nn::SequentialImpl {
                              .stride(1)
                              .with_bias(false)));
    push_back(
-        "pool", torch::nn::Functional(torch::avg_pool2d, 2, 2, 0, false, true));
+        "pool", torch::nn::Functional([](torch::Tensor input) { return torch::avg_pool2d(input, 2, 2, 0, false, true); }));
  }
  torch::Tensor forward(torch::Tensor x) {