tddandroid's picture
Create get_device
ed66a8c verified
raw
history blame contribute delete
75 Bytes
from setup_device import get_device
device = get_device()
model.to(device)