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

Merge pull request #42 from ixaxaar/quickies

Browse Source 
Fixes #36 and #39
This commit is contained in:
ixaxaar 2019-04-05 15:24:15 +05:30 committed by GitHub
commit 1db78511fe
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
dnc/sparse_temporal_memory.py
View File

@ -8,7 +8,6 @@ import torch.nn.functional as F
import numpy as np
import math
from .flann_index import FLANNIndex
from .util import *
import time

6
dnc/util.py
View File

@ -29,7 +29,7 @@ def cuda(x, grad=False, gpu_id=-1):
t.requires_grad=grad
return t
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
return t
@ -40,7 +40,7 @@ def cudavec(x, grad=False, gpu_id=-1):
t.requires_grad = grad
return t
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
return t
@ -51,7 +51,7 @@ def cudalong(x, grad=False, gpu_id=-1):
t.requires_grad = grad
return t
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
return t

2
setup.py
View File

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