X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=lazy_linear.py;h=4599f00e1fc4f36844e69a918d6b4ebeb6e883a3;hb=2db4624955ad2a1c29f7632f30ac217c045638cf;hp=7c9e398a66b16dcc025471b59b6ac874e2a6d5f3;hpb=7443d768bc437889659ba3ed737297f90fe1922e;p=pytorch.git 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),