projects
/
pytorch.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Cleaning up.
[pytorch.git]
/
ae_size.py
diff --git
a/ae_size.py
b/ae_size.py
index
25ecc4a
..
8afb101
100755
(executable)
--- a/
ae_size.py
+++ b/
ae_size.py
@@
-12,9
+12,9
@@
def minimal_input_size(w, layer_specs):
return w
else:
kernel_size, stride = layer_specs[0]
return w
else:
kernel_size, stride = layer_specs[0]
-
w = math.ceil((w - kernel_size) / stride
) + 1
-
w = minimal_input_size(w
, layer_specs[1:])
- return
int((w - 1) * stride + kernel_size)
+
v = int(math.ceil((w - kernel_size) / stride)
) + 1
+
v = minimal_input_size(v
, layer_specs[1:])
+ return
(v - 1) * stride + kernel_size
######################################################################
######################################################################
@@
-22,9
+22,10
@@
def minimal_input_size(w, layer_specs):
if __name__ == "__main__":
if __name__ == "__main__":
- layer_specs = [ (1
1
, 5), (5, 4), (3, 2), (3, 2) ]
+ layer_specs = [ (1
7
, 5), (5, 4), (3, 2), (3, 2) ]
layers = []
layers = []
+
for kernel_size, stride in layer_specs:
layers.append(nn.Conv2d(1, 1, kernel_size, stride))
for kernel_size, stride in layer_specs:
layers.append(nn.Conv2d(1, 1, kernel_size, stride))