-#!/usr/bin/env python-for-pytorch
+#!/usr/bin/env python
from torch import nn, Tensor
-##########
+######################################################################
class LazyLinear(nn.Module):
return self.core(x)
-##########
+######################################################################
model = nn.Sequential(nn.Conv2d(1, 8, kernel_size = 5),
nn.ReLU(inplace = True),