Cuda Issue while training.

something678 / TodKat

Transformer encoder-decoder for emotion detection in dialogues

MIT License

57 stars 12 forks source link

Cuda Issue while training. #1

Closed ShreyPandit closed 3 years ago

ShreyPandit commented 3 years ago

Followed the steps as mentioned in the ReadMe. Still facing error while training the IEMOCAP model. Some CPU/GPU issue I suppose. Would be great if it could be resolved. Attaching the pic of the error.

ShreyPandit commented 3 years ago

Worked out when replaced the cuda:0 to cuda everywhere in the code.