0f7823cb6a27b5d7dfab56f06a6601bf2957594b,skorch/net.py,NeuralNet,__setstate__,#NeuralNet#Any#,1320

Before Change

            with tempfile.SpooledTemporaryFile() as f:
                f.write(dump)
                f.seek(0)
                if (
                        uses_cuda(state["device"]) and
                        not torch.cuda.is_available()
                ):
                    disable_cuda = True
                    val = torch.load(
                        f, map_location=lambda storage, loc: storage)
                else:
                    val = torch.load(f)
            state[key] = val
        if disable_cuda:
            warnings.warn(
                "Model configured to use CUDA but no CUDA devices "

After Change

                        not torch.cuda.is_available())
        load_kwargs = {}
        if disable_cuda:
            load_kwargs = {"map_location": lambda store, loc: store}

        with tempfile.SpooledTemporaryFile() as f:
            f.write(state["cuda_dependent_attributes_"])
            f.seek(0)

Instances