projects
/
pytorch.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Update.
[pytorch.git]
/
ae_size.py
diff --git
a/ae_size.py
b/ae_size.py
index
067a7fa
..
49f4a20
100755
(executable)
--- a/
ae_size.py
+++ b/
ae_size.py
@@
-11,8
+11,9
@@
from torch import Tensor
######################################################################
######################################################################
+
def minimal_input_size(w, layer_specs):
def minimal_input_size(w, layer_specs):
- assert w > 0,
'The input is too small'
+ assert w > 0,
"The input is too small"
if layer_specs == []:
return w
else:
if layer_specs == []:
return w
else:
@@
-21,13
+22,13
@@
def minimal_input_size(w, layer_specs):
v = minimal_input_size(v, layer_specs[1:])
return (v - 1) * stride + kernel_size
v = minimal_input_size(v, layer_specs[1:])
return (v - 1) * stride + kernel_size
+
######################################################################
# Dummy test
if __name__ == "__main__":
######################################################################
# Dummy test
if __name__ == "__main__":
-
- layer_specs = [ (17, 5), (5, 4), (3, 2), (3, 2) ]
+ layer_specs = [(17, 5), (5, 4), (3, 2), (3, 2)]
layers = []
layers = []