File "example_save_torch_script.py", line 62, in <module>
    new_scripted_model = torch.jit.load("script_model.pt")
  File "/home/agis/anaconda3/envs/fastai-v1/lib/python3.6/site-packages/torch/jit/__init__.py", line 98, in load
    torch._C.import_ir_module(module_lookup, f)
RuntimeError: it != value_type_map_.end() ASSERT FAILED at /opt/conda/conda-bld/pytorch-nightly_1538471633750/work/torch/csrc/jit/import.cpp:281, please report a bug to PyTorch. (buildIntermediateValue at /opt/conda/conda-bld/pytorch-nightly_1538471633750/work/torch/csrc/jit/import.cpp:281)

import torch

class MyModuleS(torch.jit.ScriptModule):
    def __init__(self, N, M):
        super(MyModuleS, self).__init__()
        self.weight = torch.nn.Parameter(torch.rand(N, M))

    @torch.jit.script_method
    def forward(self, input: torch.Tensor):
        if bool(input.sum() > 0):
            output = self.weight.mv(input)
        else:
            output = self.weight + input
        return output


my_script_module = MyModuleS(200, 200)

test_input = torch.rand(200)

output = my_script_module(test_input)
print(my_script_module.graph)
print("scripted model: ", output.size())

my_script_module.save("script_model.pt")


new_scripted_model = torch.jit.load("script_model.pt")  # crash happens here
output = new_scripted_model(test_input)
print(output)

wget https://raw.githubusercontent.com/pytorch/pytorch/master/torch/utils/collect_env.py
# For security purposes, please check the contents of collect_env.py before running it.
python collect_env.py