pytorch-NeuCom

pytorch-NeuCom copied to clipboard

TypeError: norm received an invalid combination of arguments - got (int, bool), but expected one of:

2

(VENVpytorch2) mldl@mldlUB1604:~/ub16_prj/pytorch-NeuCom/tasks/Copy$ python train.py Using CUDA. Iteration 0/100000Traceback (most recent call last): File "train.py", line 177, in output, _ = ncomputer(input_data) File "/home/mldl/ub16_prj/VENV_host/VENV_pytorch2/VENVpytorch2/local/lib/python2.7/site-packages/torch/nn/modules/module.py", line 206, in __call__ result = self.forward(*input,...

SeekPoint

backward through cumprod

12

Hey, In function ```get_allocation_weight()``` in ```memory.py```, ```cumprod``` is used, but as far as I know, this operation does not have autograd support yet and is still in PR: [https://github.com/pytorch/pytorch/pull/1439](url), so...

jingweiz

Compare with DeepMind's implementation

2

Hi @ypxie, I just saw DeepMind released their Tensorflow/Sonnet implementation of the DNC, https://github.com/deepmind/dnc I was wondering if you're interested in re-looking at this project to see if we can...

ajaytalati