bf1495ab27933de58429bb643321fc3e21933d19,inferno/net.py,NeuralNet,__setstate__,#NeuralNet#Any#,734

Before Change

    def __setstate__(self, state):
        if "module_" in state:
            module_dump = state.pop("module_")
            module_ = pickle.loads(module_dump)
            state["module_"] = module_
        self.__dict__.update(state)

    def save_params(self, f):

After Change

        if "module_" in state:
            module_dump = state.pop("module_")

            with tempfile.SpooledTemporaryFile() as f:
                f.write(module_dump)
                f.seek(0)
                if state["use_cuda"] and not torch.cuda.is_available():
                    warnings.warn(
                        "Model configured to use CUDA but no CUDA devices "
                        "available. Loading on CPU instead.",
                        ResourceWarning)
                    module_ = torch.load(f, map_location=lambda storage, loc: storage)
                else:
                    module_ = torch.load(f)
            state["module_"] = module_
        self.__dict__.update(state)

    def save_params(self, f):

Instances