projects
/
pytorch.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
|
inline
| side by side (from parent 1:
0a30c67
)
Update.
author
Francois Fleuret
<francois@fleuret.org>
Sat, 9 Jun 2018 12:15:53 +0000
(14:15 +0200)
committer
Francois Fleuret
<francois@fleuret.org>
Sat, 9 Jun 2018 12:15:53 +0000
(14:15 +0200)
ae_size.py
patch
|
blob
|
history
diff --git
a/ae_size.py
b/ae_size.py
index
7183937
..
48dc2af
100755
(executable)
--- a/
ae_size.py
+++ b/
ae_size.py
@@
-25,11
+25,11
@@
if __name__ == "__main__":
layer_specs = [ (11, 5), (5, 2), (3, 2), (3, 2) ]
layers = []
layer_specs = [ (11, 5), (5, 2), (3, 2), (3, 2) ]
layers = []
- for
l
in layer_specs:
- layers.append(nn.Conv2d(1, 1,
l[0], l[1]
))
+ for
kernel_size, stride
in layer_specs:
+ layers.append(nn.Conv2d(1, 1,
kernel_size, stride
))
- for
l
in reversed(layer_specs):
- layers.append(nn.ConvTranspose2d(1, 1,
l[0], l[1]
))
+ for
kernel_size, stride
in reversed(layer_specs):
+ layers.append(nn.ConvTranspose2d(1, 1,
kernel_size, stride
))
m = nn.Sequential(*layers)
m = nn.Sequential(*layers)