คำถามในหัวข้อ 'pytorch-dataloader'
PyTorch: ขณะโหลดข้อมูลแบบแบตช์โดยใช้ Dataloader วิธีถ่ายโอนข้อมูลไปยัง GPU โดยอัตโนมัติ
หากเราใช้คลาส Dataset และ Dataloader รวมกัน (ดังที่แสดงด้านล่าง) ฉันจะต้องโหลดข้อมูลลงใน GPU อย่างชัดเจนโดยใช้ .to() หรือ .cuda() มีวิธีสั่งให้ dataloader ทำโดยอัตโนมัติ/โดยปริยายหรือไม่?
รหัสเพื่อทำความเข้าใจ/สร้างสถานการณ์จำลอง:
from...
545 มุมมอง
schedule
28.10.2023