Update.

[pytorch.git] / lazy_linear.py

diff --git a/lazy_linear.py b/lazy_linear.py
index 7c9e398..4599f00 100755 (executable)
--- 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),