Searched refs:MyPickledModule (Results 1 – 1 of 1) sorted by relevance
115 class MyPickledModule(torch.nn.Module): class6535 MyPickledModule(torch.randn([2, 2]))6543 model = MyPickledModule(torch.randn([2, 2]))