Thesis/pytorch-dnc
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix #36

Browse Source
This commit is contained in:
Russi Chatterjee 2019-04-05 15:22:48 +05:30
parent 4266c2e7aa
commit 3e477fcf18
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
dnc/util.py
View File

@ -29,7 +29,7 @@ def cuda(x, grad=False, gpu_id=-1):
t.requires_grad=grad t.requires_grad=grad
return t return t
else: else:
t = T.FloatTensor(x.pin_memory()).cuda(gpu_id, async=True) t = T.FloatTensor(x.pin_memory()).cuda(gpu_id)
t.requires_grad=grad t.requires_grad=grad
return t return t
@ -40,7 +40,7 @@ def cudavec(x, grad=False, gpu_id=-1):
t.requires_grad = grad t.requires_grad = grad
return t return t
else: else:
t = T.Tensor(T.from_numpy(x).pin_memory()).cuda(gpu_id, async=True) t = T.Tensor(T.from_numpy(x).pin_memory()).cuda(gpu_id)
t.requires_grad = grad t.requires_grad = grad
return t return t
@ -51,7 +51,7 @@ def cudalong(x, grad=False, gpu_id=-1):
t.requires_grad = grad t.requires_grad = grad
return t return t
else: else:
t = T.LongTensor(T.from_numpy(x.astype(np.long)).pin_memory()).cuda(gpu_id, async=True) t = T.LongTensor(T.from_numpy(x.astype(np.long)).pin_memory()).cuda(gpu_id)
t.requires_grad = grad t.requires_grad = grad
return t return t

2
setup.py
View File

@ -22,7 +22,7 @@ with open(path.join(here, 'README.rst'), encoding='utf-8') as f:
setup( setup(
name='dnc', name='dnc',
version='1.0.0', version='1.0.1',
description='Differentiable Neural Computer, for Pytorch', description='Differentiable Neural Computer, for Pytorch',
long_description=long_description, long_description=long_description,