From: Francois Fleuret Date: Thu, 10 May 2018 09:21:07 +0000 (+0200) Subject: Update. X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pytorch.git;a=commitdiff_plain;h=7b534d513dd1bdad208f2b59baf2d47979f90663 Update. --- diff --git a/lazy_linear.py b/lazy_linear.py index 7c9e398..4599f00 100755 --- a/lazy_linear.py +++ b/lazy_linear.py @@ -1,8 +1,8 @@ -#!/usr/bin/env python-for-pytorch +#!/usr/bin/env python from torch import nn, Tensor -########## +###################################################################### class LazyLinear(nn.Module): @@ -23,7 +23,7 @@ class LazyLinear(nn.Module): return self.core(x) -########## +###################################################################### model = nn.Sequential(nn.Conv2d(1, 8, kernel_size = 5), nn.ReLU(inplace = True),