This commit is contained in:
Morvan Zhou
2017-05-23 10:37:51 +10:00
committed by Morvan Zhou
parent 30c0070497
commit d87529ed89
10 changed files with 17 additions and 10 deletions

View File

@ -32,4 +32,18 @@ net2 = torch.nn.Sequential(
print(net1) # net1 architecture
print(net2) # net2 architecture
"""
Net (
(hidden): Linear (1 -> 10)
(predict): Linear (10 -> 1)
)
"""
print(net2) # net2 architecture
"""
Sequential (
(0): Linear (1 -> 10)
(1): ReLU ()
(2): Linear (10 -> 1)
)
"""