From 0c2da9bc90d21bdbcbcbeeb0a070653cbc3c25cc Mon Sep 17 00:00:00 2001 From: Francois Fleuret Date: Sun, 6 Jun 2021 16:08:03 +0200 Subject: [PATCH] Update. --- conv_chain.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/conv_chain.py b/conv_chain.py index 2d5af8e..d10798f 100755 --- a/conv_chain.py +++ b/conv_chain.py @@ -1,7 +1,9 @@ #!/usr/bin/env python -import torch -from torch import nn +# Any copyright is dedicated to the Public Domain. +# https://creativecommons.org/publicdomain/zero/1.0/ + +# Written by Francois Fleuret ###################################################################### @@ -26,13 +28,16 @@ def conv_chain(input_size, output_size, depth, cond): if __name__ == "__main__": + import torch + from torch import nn + # Example c = conv_chain( input_size = 64, output_size = 8, depth = 5, # We want kernels smaller than 4, strides smaller than the - # kernels, and stride of 1 except in the two last layers + # kernels, and strides of 1 except in the two last layers cond = lambda d, k, s: k <= 4 and s <= k and (s == 1 or d <= 2) ) -- 2.39.5