Added conditional check for use_cuda before using torch.cuda
#42
| Job | Run time |
|---|---|
| 0s |
use_cuda before using torch.cuda
#42
| Job | Run time |
|---|---|
| 0s |