RuntimeError: The following operation failed in the TorchScript interpreter. RuntimeError: Unsupported value kind: Tensor

[martinjankowiak]

torch jit is finicky and probably doesn’t like some aspect of PyroModule. i suggest you write your model as a simple python function as is done e.g. here