Lines Matching refs:MyModule

49 class MyModule(torch.nn.Module):  class
67             + [MyModule(10000, 10000)]
68             + [MyModule(10000, 1000)]
69             + [MyModule(1000, 1000)]
70             + [MyModule(1000, 1000)]
71             + [MyModule(1000, 1000)]
72             + [MyModule(1000, 1000)]
73             + [MyModule(1000, 1000)]
74             + [MyModule(1000, 1000)]
75             + [MyModule(1000, 1000)]
130     "toy_model": (MyModule,),