frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-11-01 04:54:55 +08:00
Code Issues Packages Projects Releases Wiki Activity
17,136 Commits 5,309 Branches 1,478 Tags
dfcd7b0185d479f186ddb100c761f4df3495f8e8
Commit Graph

3 Commits

Author SHA1 Message Date
Will Feng
6ebfbdf4c6 Add named submodule support to nn::Sequential (#17552) 
Summary:
Previously, we were not able to assign names to `nn::Sequential`'s submodules. This PR adds this feature to match the Python API. Example use:
```cpp
Sequential sequential(named_submodule({
      {"linear", Linear(10, 3)},
      {"conv2d", Conv2d(1, 2, 3)},
      {"dropout", Dropout(0.5)},
      {"batchnorm", BatchNorm(5)},
      {"embedding", Embedding(4, 10)},
      {"lstm", LSTM(4, 5)}
}));
```

It also enables loading parameters of Python `nn.Sequential` module with custom submodules names into C++ frontend, unblocking https://github.com/pytorch/vision/pull/728#issuecomment-466661344.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/17552

Differential Revision: D14246834

Pulled By: yf225

fbshipit-source-id: 3030b5c5d68f6dd5d3e37ac4b4f98dc6d6d9ba72
 2019-03-29 13:06:29 -07:00
Peter Goldsborough
49fe678fec Add variable_factories.h to cppdocs (#14381) 
Summary:
This will document `torch::from_blob` and such.

soumith ezyang
Pull Request resolved: https://github.com/pytorch/pytorch/pull/14381

Differential Revision: D13216560

Pulled By: goldsborough

fbshipit-source-id: 112f60e45e4d38a8a9983fa71e9cc56bc1a73465
 2018-11-27 10:13:23 -08:00
sven
e585f2fb48 Polish CPP docs, Minor Python Docs Fixes (#11722) 
Differential Revision: D9919120

Pulled By: goldsborough

fbshipit-source-id: bf14cbe4ab79524495957cb749828046af864aab
 2018-09-18 14:55:57 -07:00