Searched defs:optim2 (Results 1 – 2 of 2) sorted by relevance
43 def _test_optim_state_save_load(self, model1, optim1, model2, optim2) -> None: argument
101 auto optim2 = OptimizerClass(model2->parameters(), options); in test_serialize_optimizer() local492 auto optim2 = torch::optim::SGD( in TEST() local