YYuanAnyVision / mxnet_center_loss

implement center loss operator for mxnet
193 stars 93 forks source link

run vis.py 时出现错误 #2

Open Trangle opened 7 years ago

Trangle commented 7 years ago

C:\WinPython\python-2.7.10.amd64\python.exe D:/MyCoding/DeepLearning/test_example/mxnet/example/center_loss/vis.py feature_extractor loaded [01:15:43] D:\MyCoding\DeepLearning\OpenSource\mxnet\src\io\iter_mnist.cc:94: MNISTIter: load 60000 images, shuffle=1, shape=(100,1,28,28) [01:15:44] D:\MyCoding\DeepLearning\OpenSource\mxnet\src\io\iter_mnist.cc:94: MNISTIter: load 10000 images, shuffle=1, shape=(100,1,28,28) extracting feature [01:15:44] D:\MyCoding\DeepLearning\OpenSource\mxnet\dmlc-core\include\dmlc/logging.h:235: [01:15:44] D:\MyCoding\DeepLearning\OpenSource\mxnet\src\symbol\symbol.cc:155: Symbol.InferShapeKeyword argument name softmax_label not found. Candidate arguments: [0]data [1]convolution0_weight [2]convolution0_bias [3]convolution1_weight [4]convolution1_bias [5]fullyconnected0_weight [6]fullyconnected0_bias [7]embedding_weight [8]embedding_bias

Traceback (most recent call last): File "D:/MyCoding/DeepLearning/test_example/mxnet/example/center_loss/vis.py", line 73, in main() File "D:/MyCoding/DeepLearning/test_example/mxnet/example/center_loss/vis.py", line 55, in main preds = feature_extractor.predict( i.data[0] ) File "C:\WinPython\python-2.7.10.amd64\lib\site-packages\mxnet-0.7.0-py2.7.egg\mxnet\model.py", line 608, in predict self._init_predictor(data_shapes, type_dict) File "C:\WinPython\python-2.7.10.amd64\lib\site-packages\mxnet-0.7.0-py2.7.egg\mxnet\model.py", line 532, in _init_predictor self.ctx[0], grad_req='null', type_dict=type_dict, dict(input_shapes)) File "C:\WinPython\python-2.7.10.amd64\lib\site-packages\mxnet-0.7.0-py2.7.egg\mxnet\symbol.py", line 676, in simple_bind argshapes, , aux_shapes = self.infer_shape(kwargs) File "C:\WinPython\python-2.7.10.amd64\lib\site-packages\mxnet-0.7.0-py2.7.egg\mxnet\symbol.py", line 458, in infer_shape return self._infer_shape_impl(False, *args, **kwargs) File "C:\WinPython\python-2.7.10.amd64\lib\site-packages\mxnet-0.7.0-py2.7.egg\mxnet\symbol.py", line 518, in _infer_shape_impl ctypes.byref(complete))) File "C:\WinPython\python-2.7.10.amd64\lib\site-packages\mxnet-0.7.0-py2.7.egg\mxnet\base.py", line 77, in check_call raise MXNetError(py_str(_LIB.MXGetLastError())) mxnet.base.MXNetError: [01:15:44] D:\MyCoding\DeepLearning\OpenSource\mxnet\src\symbol\symbol.cc:155: Symbol.InferShapeKeyword argument name softmax_label not found. Candidate arguments: [0]data [1]convolution0_weight [2]convolution0_bias [3]convolution1_weight [4]convolution1_bias [5]fullyconnected0_weight [6]fullyconnected0_bias [7]embedding_weight [8]embedding_bias