Expected integer literal for index

I opened this issue becasue torch.jit.script is not working with hrnet