graykode / graykode/nlp-tutorial
Transformer/Transformer(Greedy_decoder)-Torch.py on gpu
Offen
- Vorherrschende Sprache
- Jupyter Notebook
- Sterne
- 14.9k
- Forks
- 3.9k
- PR-Merge-Kennzahlen
- Keine gemergten PRs in 30 T.
Beschreibung
Hello, I want to put the Transformer (Greedy_decoder)-Torch.py code on the gpu, using model=model.to(device), input_data also to (device), but the error still appears "Expected object of backend CUDA but backend CPU for argument #2 'mat2”
Beitragsleitfaden
Bewertung
Dieses Issue wurde noch nicht bewertet.