Searched defs:optim1 (Results 1 – 2 of 2) sorted by relevance
99 auto optim1 = OptimizerClass( in test_serialize_optimizer() local490 auto optim1 = torch::optim::SGD( in TEST() local545 auto optim1 = torch::optim::Adagrad( in TEST() local593 auto optim1 = torch::optim::SGD( in TEST() local642 auto optim1 = torch::optim::Adam( in TEST() local701 auto optim1 = torch::optim::AdamW( in TEST() local760 auto optim1 = torch::optim::RMSprop(model1_params, options); in TEST() local828 auto optim1 = in TEST() local
43 def _test_optim_state_save_load(self, model1, optim1, model2, optim2) -> None: argument