File "E:\code\FlowChain-ICCV2023-main\src\models\TP\mgcvae.py", line 542, in encode_node_history
return outputs[torch.arange(first_history_indices.shape[0]), last_index_per_sequence]
IndexError: tensors used as indices must be long, byte or bool tensors
File "E:\code\FlowChain-ICCV2023-main\src\models\TP\mgcvae.py", line 542, in encode_node_history return outputs[torch.arange(first_history_indices.shape[0]), last_index_per_sequence] IndexError: tensors used as indices must be long, byte or bool tensors