search

tryolabs / luminoth

Deep Learning toolkit for Computer Vision.
https://tryolabs.com
BSD 3-Clause "New" or "Revised" License
2.4k stars 399 forks source link

TypeError: softmax() got an unexpected keyword argument 'axis' #169

Closed Phillylyu closed 6 years ago

Phillylyu commented 6 years ago

I am using the latest source code to run the trainning . I suffered the following error: File "/root/luminoth/luminoth/models/fasterrcnn/rcnn.py", line 221, in _build cls_prob = tf.nn.softmax(cls_score, axis=1) TypeError: softmax() got an unexpected keyword argument 'axis'

My Environment: TF 1.3+Ubutu14.04 +Anaconda 3+python 3.5

nagitsu commented 6 years ago

Try using the latest Tensorflow version, as the dim argument of tf.nn.softmax was deprecated and changed for axis.

We'll update the dependencies accordingly, as we seem to have missed it.