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]
/
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
from torch import nn, Tensor
-##########
+##########
############################################################
class LazyLinear(nn.Module):
class LazyLinear(nn.Module):
@@
-23,7
+23,7
@@
class LazyLinear(nn.Module):
return self.core(x)
return self.core(x)
-##########
+##########
############################################################
model = nn.Sequential(nn.Conv2d(1, 8, kernel_size = 5),
nn.ReLU(inplace = True),
model = nn.Sequential(nn.Conv2d(1, 8, kernel_size = 5),
nn.ReLU(inplace = True),