• Michael Suo's avatar
    [jit] don't try to set training after ScriptModule has been initialized. · 7c404fa5
    Michael Suo 提交于
    Now when initializing a ScriptModule during the torch.jit.load()
    process, there is already a cpp module backing the thing. That means
    that setting training will overwrite whatever the initialized
    ScriptModule had.
    
    This PR splits apart the common "set up internal state" part of the
    Module __init__ and calls that from ScriptModule.__init__ and
    Module.__init__, leaving the "nn.Module-specific" part (setting
    `self.training`) for the nn.Module __init__
    
    ghstack-source-id: 9b2ba8a15c43cf230363e4cd10ba4ad3ac4931f7
    Pull Request resolved: https://github.com/pytorch/pytorch/pull/23680
    7c404fa5
名称
最后提交
最后更新
.circleci Loading commit data...
.ctags.d Loading commit data...
.github Loading commit data...
.jenkins Loading commit data...
aten Loading commit data...
benchmarks Loading commit data...
binaries Loading commit data...
c10 Loading commit data...
caffe2 Loading commit data...
cmake Loading commit data...
docker Loading commit data...
docs Loading commit data...
modules Loading commit data...
scripts Loading commit data...
submodules Loading commit data...
test Loading commit data...
third_party Loading commit data...
tools Loading commit data...
torch Loading commit data...
.clang-format Loading commit data...
.clang-tidy Loading commit data...
.dockerignore Loading commit data...
.flake8 Loading commit data...
.gitattributes Loading commit data...
.gitignore Loading commit data...
.gitmodules Loading commit data...
.travis.aten.yml Loading commit data...
.travis.yml Loading commit data...
CITATION Loading commit data...
CMakeLists.txt Loading commit data...
CODEOWNERS Loading commit data...
CONTRIBUTING.md Loading commit data...
LICENSE Loading commit data...
Makefile Loading commit data...
NOTICE Loading commit data...
README.md Loading commit data...
azure-pipelines.yml Loading commit data...
mypy-README.md Loading commit data...
mypy-files.txt Loading commit data...
mypy.ini Loading commit data...
requirements.txt Loading commit data...
setup.py Loading commit data...
ubsan.supp Loading commit data...