Searched refs:NestedSequentialModel (Results 1 – 3 of 3) sorted by relevance
133 class NestedSequentialModel: class in TestFSDPWrap442 seq = TestFSDPWrap.NestedSequentialModel.get_model(cuda=True)446 TestFSDPWrap.NestedSequentialModel.verify_model_all_wrapped(self, model)604 sequential = TestFSDPWrap.NestedSequentialModel.get_model(cuda=False)614 TestFSDPWrap.NestedSequentialModel.verify_model(self, model)735 sequential = TestFSDPWrap.NestedSequentialModel.get_model(747 TestFSDPWrap.NestedSequentialModel.verify_model(self, model)765 sequential = TestFSDPWrap.NestedSequentialModel.get_model(cuda=False)790 sequential = TestFSDPWrap.NestedSequentialModel.get_model(cuda=False)
18 NestedSequentialModel,73 NestedSequentialModel(torch.device("cuda"))
83 class NestedSequentialModel(nn.Module): class