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

Before Change

                        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:

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)
            cuda_attrs = torch.load(f, **load_kwargs)

        set_cuda_attrs = {}
        state.update(cuda_attrs)
        for key in self.cuda_dependent_attributes_:
            if key not in cuda_attrs:
                continue
            set_cuda_attrs[key] = state.pop(key)

Instances