Searched refs:_create_scaling_models_optimizers (Results 1 – 3 of 3) sorted by relevance
246 def _create_scaling_models_optimizers(device="cuda", optimizer_ctor=torch.optim.SGD, optimizer_kwar… function274 return _create_scaling_models_optimizers(
21 _create_scaling_models_optimizers,1216 ) = _create_scaling_models_optimizers(device=dev1)
61 _create_scaling_case, _create_scaling_models_optimizers)6114 _create_scaling_models_optimizers(device.type)