icefall/.mypy_cache/3.9/torch/optim/swa_utils.data.json

{".class": "MypyFile", "_fullname": "torch.optim.swa_utils", "is_partial_stub_package": false, "is_stub": true, "names": {".class": "SymbolTable", "Any": {".class": "SymbolTableNode", "cross_ref": "typing.Any", "kind": "Gdef", "module_hidden": true, "module_public": false}, "AveragedModel": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["torch.nn.modules.module.Module"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.optim.swa_utils.AveragedModel", "name": "AveragedModel", "type_vars": []}, "flags": [], "fullname": "torch.optim.swa_utils.AveragedModel", "metaclass_type": null, "metadata": {}, "module_name": "torch.optim.swa_utils", "mro": ["torch.optim.swa_utils.AveragedModel", "torch.nn.modules.module.Module", "builtins.object"], "names": {".class": "SymbolTable", "__init__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1], "arg_names": ["self", "model", "device", "avg_fn"], "flags": [], "fullname": "torch.optim.swa_utils.AveragedModel.__init__", "name": "__init__", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1], "arg_names": ["self", "model", "device", "avg_fn"], "arg_types": ["torch.optim.swa_utils.AveragedModel", "torch.nn.modules.module.Module", {".class": "UnionType", "items": ["builtins.int", "torch._C.device"]}, {".class": "CallableType", "arg_kinds": [0, 0, 0], "arg_names": [null, null, null], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", "builtins.int"], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": null, "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "__init__ of AveragedModel", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "update_parameters": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["self", "model"], "flags": [], "fullname": "torch.optim.swa_utils.AveragedModel.update_parameters", "name": "update_parameters", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["self", "model"], "arg_types": ["torch.optim.swa_utils.AveragedModel", "torch.nn.modules.module.Module"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "update_parameters of AveragedModel", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "Callable": {".class": "SymbolTableNode", "cross_ref": "typing.Callable", "kind": "Gdef", "module_hidden": true, "module_public": false}, "Iterable": {".class": "SymbolTableNode", "cross_ref": "typing.Iterable", "kind": "Gdef", "module_hidden": true, "module_public": false}, "List": {".class": "SymbolTableNode", "cross_ref": "typing.List", "kind": "Gdef", "module_hidden": true, "module_public": false}, "Module": {".class": "SymbolTableNode", "cross_ref": "torch.nn.modules.module.Module", "kind": "Gdef", "module_hidden": true, "module_public": false}, "Optimizer": {".class": "SymbolTableNode", "cross_ref": "torch.optim.optimizer.Optimizer", "kind": "Gdef", "module_hidden": true, "module_public": false}, "Optional": {".class": "SymbolTableNode", "cross_ref": "typing.Optional", "kind": "Gdef", "module_hidden": true, "module_public": false}, "SWALR": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["torch.optim.lr_scheduler._LRScheduler"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.optim.swa_utils.SWALR", "name": "SWALR", "type_vars": []}, "flags": [], "fullname": "torch.optim.swa_utils.SWALR", "metaclass_type": null, "metadata": {}, "module_name": "torch.optim.swa_utils", "mro": ["torch.optim.swa_utils.SWALR", "torch.optim.lr_scheduler._LRScheduler", "builtins.object"], "names": {".class": "SymbolTable", "__init__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 0, 0, 1], "arg_names": ["self", "optimizer", "swa_lr", "anneal_epochs", "anneal_strategy", "last_epoch"], "flags": [], "fullname": "torch.optim.swa_utils.SWALR.__init__", "name": "__init__", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 0, 0, 1], "arg_names": ["self", "optimizer", "swa_lr", "anneal_epochs", "anneal_strategy", "last_epoch"], "arg_types": ["torch.optim.swa_utils.SWALR", "torch.optim.optimizer.Optimizer", "builtins.float", "builtins.int", "builtins.str", "builtins.int"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "__init__ of SWALR", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "Tensor": {".class": "SymbolTableNode", "cross_ref": "torch._tensor.Tensor", "kind": "Gdef", "module_hidden": true, "module_public": false}, "Union": {".class": "SymbolTableNode", "cross_ref": "typing.Union", "kind": "Gdef", "module_hidden": true, "module_public": false}, "_LRScheduler": {".class": "SymbolTableNode", "cross_ref": "torch.optim.lr_scheduler._LRScheduler", "kind": "Gdef", "module_hidden": true, "module_public": false}, "__doc__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.optim.swa_utils.__doc__", "name": "__doc__", "type": "builtins.str"}}, "__file__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.optim.swa_utils.__file__", "name": "__file__", "type": "builtins.str"}}, "__name__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.optim.swa_utils.__name__", "name": "__name__", "type": "builtins.str"}}, "__package__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.optim.swa_utils.__package__", "name": "__package__", "type": "builtins.str"}}, "device": {".class": "SymbolTableNode", "cross_ref": "torch._C.device", "kind": "Gdef", "module_hidden": true, "module_public": false}, "update_bn": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1], "arg_names": ["loader", "model", "device"], "flags": [], "fullname": "torch.optim.swa_utils.update_bn", "name": "update_bn", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1], "arg_names": ["loader", "model", "device"], "arg_types": [{".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}], "type_ref": "typing.Iterable"}, "torch.nn.modules.module.Module", {".class": "UnionType", "items": ["builtins.int", "torch._C.device"]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "update_bn", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}}, "path": "/home/patrick/anaconda3/lib/python3.9/site-packages/torch/optim/swa_utils.pyi"}