Open NicolasBFR opened 6 months ago
Tensors and models now loaded on CPU if CUDA not available. (cuda() instructions replaced with to(device))
cuda()
to(device)
Tensors and models now loaded on CPU if CUDA not available. (
cuda()
instructions replaced withto(device)
)