graykode / graykode/nlp-tutorial
Transformer/Transformer(Greedy_decoder)-Torch.py on gpu
Đang mở
- Ngôn ngữ chính
- Jupyter Notebook
- Star
- 14.9k
- Fork
- 3.9k
- Chỉ số merge pull request
- Không có pull request nào được merge trong 30 ngày
Mô tả
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”
Hướng dẫn đóng góp
Hướng nghiên cứu
Start with Transformer/Transformer(Greedy_decoder)-Torch.py and reproduce the reported CUDA-versus-CPU mat2 error. Trace the devices of model, input_data, and the tensors involved in the failing operation. Done means the example runs on the GPU without a backend mismatch.
Do mô hình lập chỉ mục viết ra từ nội dung của issue.
Đánh giá
- Công nghệ
- python
- Lĩnh vực
- machine-learning
- Loại issue
- Lỗi
- Độ khó
- 3/5
- Thời gian dự kiến
- 1-2 ngày
- Mức độ hoạt động
- Đình trệ
- Độ rõ ràng
- Cần làm rõ
- Mức phù hợp với người mới
- 25/100