+ def named_parameters(self, memo=None, prefix=""):
+ assert self.core is not None, "Parameters not yet defined"
+ return super().named_parameters(memo, prefix)
+
+
+######################################################################
+
+if __name__ == "__main__":
+ model = nn.Sequential(
+ nn.Conv2d(3, 8, kernel_size=5),
+ nn.ReLU(inplace=True),
+ LazyLinear(128),
+ nn.ReLU(inplace=True),
+ nn.Linear(128, 10),
+ )