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]
/
conv_chain.py
diff --git
a/conv_chain.py
b/conv_chain.py
index
85ae4fc
..
184e06b
100755
(executable)
--- a/
conv_chain.py
+++ b/
conv_chain.py
@@
-31,7
+31,9
@@
if __name__ == "__main__":
c = conv_chain(
input_size = 64, output_size = 8,
depth = 5,
c = conv_chain(
input_size = 64, output_size = 8,
depth = 5,
- cond = lambda d, k, s: k <= 4 and s <= k and (s == 1 or d < 3)
+ # We want kernels smaller than 4, strides smaller than the
+ # kernels, and stride of 1 except in the two last layers
+ cond = lambda d, k, s: k <= 4 and s <= k and (s == 1 or d <= 2)
)
x = torch.rand(1, 1, 64)
)
x = torch.rand(1, 1, 64)