Set device bc I apparently forgot to do that

This commit is contained in:
jlevine18 2019-09-23 00:01:31 -05:00 committed by GitHub
parent c5d087dada
commit dc6f896071
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -38,6 +38,8 @@ __all__ = [
import torch
#set device
device='cuda:0' if torch.cuda.is_available() else 'cpu'
#todo: document completely