Searched defs:optimizer_kwargs (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_cuda.py | 246 …te_scaling_models_optimizers(device="cuda", optimizer_ctor=torch.optim.SGD, optimizer_kwargs=None): argument
264 …ling_case(device="cuda", dtype=torch.float, optimizer_ctor=torch.optim.SGD, optimizer_kwargs=None): argument
|
| /aosp_15_r20/external/pytorch/torch/distributed/_shard/sharded_optim/ |
| H A D | api.py | 15 **optimizer_kwargs,
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_torch.py | 5852 …device, run, unskipped, skipped, atol=1e-7, optimizer_ctor=torch.optim.SGD, optimizer_kwargs=None): argument
|