Searched refs:TestDummyModel (Results 1 – 6 of 6) sorted by relevance
28 class TestDummyModel(torch.nn.Module): class78 m = TestDummyModel()79 m_dup = TestDummyModel()125 m = TestDummyModel()126 m_dup = TestDummyModel()209 m = TestDummyModel()279 m = TestDummyModel()318 m = TestDummyModel()343 m = TestDummyModel()344 m_dup = TestDummyModel()[all …]
51 class TestDummyModel(torch.nn.Module): class150 dummy_model = TestDummyModel().cuda()182 model.get_input = partial(TestDummyModel.get_input, model)425 model = TestDummyModel()435 model_2 = TestDummyModel()
30 class TestDummyModel(torch.nn.Module): class47 model = TestDummyModel().cuda()
31 class TestDummyModel(torch.nn.Module): class66 TestDummyModel() if is_even_sharded_model else TestDummyModelUneven()
292 class TestDummyModel(torch.nn.Module): class1737 model = FSDP(TestDummyModel().cuda())1849 model = FSDP(TestDummyModel().cuda())1918 model = FSDP(TestDummyModel().cuda(), use_orig_params=use_orig_params)1954 TestDummyModel().cuda(),1981 model = TestDummyModel(no_grad=True).cuda()
137 class TestDummyModel(torch.nn.Module): class1175 model = FSDP(TestDummyModel().cuda())