dialogue_gcn
dialogue_gcn copied to clipboard
Runtime error
I am trying to use dialogueGCN for establishing baseline for my model, but I cannot run the code- while executing ./scripts/iemocap.sh train
Error: RuntimeError: 'lengths' argument should be a 1D CPU int64 tensor, but got 1D cuda:1 Long tensor
Error log-
Traceback (most recent call last):
File "train.py", line 93, in
adding .cpu() to dgcn/model/SeqContext.py line 22
text_len_tensor ===>. text_len_tensor.cpu()
However it resulted in a new issue. TypeError: forward() got an unexpected keyword argument 'edge_norm'
log-
Traceback (most recent call last):
File "train.py", line 93, in
please help
Please check your PyTorch version and PyTorch Geometric version.
I am trying to use dialogueGCN for establishing baseline for my model, but I cannot run the code- while executing ./scripts/iemocap.sh train
Error: RuntimeError: 'lengths' argument should be a 1D CPU int64 tensor, but got 1D cuda:1 Long tensor
Error log- Traceback (most recent call last): File "train.py", line 93, in main(args) File "train.py", line 35, in main ret = coach.train() File "/mnt/berry/home/prakhar/dialoguegcn/conv-emotion/DialogueGCN-mianzhang/dgcn/Coach.py", line 41, in train self.train_epoch(epoch) File "/mnt/berry/home/prakhar/dialoguegcn/conv-emotion/DialogueGCN-mianzhang/dgcn/Coach.py", line 74, in train_epoch nll = self.model.get_loss(data) File "/mnt/berry/home/prakhar/dialoguegcn/conv-emotion/DialogueGCN-mianzhang/dgcn/model/DialogueGCN.py", line 59, in get_loss graph_out, features = self.get_rep(data) File "/mnt/berry/home/prakhar/dialoguegcn/conv-emotion/DialogueGCN-mianzhang/dgcn/model/DialogueGCN.py", line 43, in get_rep node_features = self.rnn(data["text_len_tensor"], data["text_tensor"]) # [batch_size, mx_len, D_g] File "/mnt/berry/home/prakhar/dialoguegcn/conv-emotion/DialogueGCN-mianzhang/cudagcn/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1051, in _call_impl return forward_call(*input, **kwargs) File "/mnt/berry/home/prakhar/dialoguegcn/conv-emotion/DialogueGCN-mianzhang/dgcn/model/SeqContext.py", line 20, in forward packed = pack_padded_sequence( File "/mnt/berry/home/prakhar/dialoguegcn/conv-emotion/DialogueGCN-mianzhang/cudagcn/lib/python3.8/site-packages/torch/nn/utils/rnn.py", line 249, in pack_padded_sequence _VF._pack_padded_sequence(input, lengths, batch_first) RuntimeError: 'lengths' argument should be a 1D CPU int64 tensor, but got 1D cuda:1 Long tensor
Maybe you should try text_len_tensor.cpu().to(torch.int64)