Torch.jit.script not working with this model

[WaterKnight1998]

I have tried to convert this model to TorchScript using torch.jit.script. However, I am getting this issue:

RuntimeError: 
Expected integer literal for index:
  File "/home/david/Documents/pry/models/archs/hrnet.py", line 228

        for i in range(self.num_branches):
            x[i] = self.branches[i](x[i])
                   ~~~~~~~~~~~~~~~ <--- HERE

        x_fuse = []

[kbegiedza]

@WaterKnight1998 Did you manage to solve it?

[nistarlwc]

same problem