NTM-tensorflow
NTM-tensorflow copied to clipboard
TypeError: softmax_loss_function() got an unexpected keyword argument 'logits'
After running 2to3 -w on all the .py files, I got an error using python3.4 running the first example in the readme (train):
[] Building a NTM model
Percent: [####################] 100.00% Finished.
[] Build a NTM model finished
[*] Reading checkpoints...
Traceback (most recent call last):
File "main.py", line 72, in
I've got the same error with Tensorflow 1.2.1. But as mentioned in Prerequisites section, it works with Tensorflow 1.1.0
This specific error can be fixed by simply changing softmax_loss_function(labels, inputs)
to softmax_loss_function(labels, logits)
in ntm.py#L15, but further errors might be raised regarding tensorflow version changes.