singa-dev mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From GitBox <...@apache.org>
Subject [GitHub] [singa] lijiansong opened a new issue #807: AssertionError for the ONNX training testcases?
Date Wed, 14 Oct 2020 07:40:12 GMT

lijiansong opened a new issue #807:
URL: https://github.com/apache/singa/issues/807


   AssertionError with the onnx testcase: <https://github.com/apache/singa/blob/master/examples/onnx/training/train.py>
   
   ```
   $ cd examples/onnx
   $ python3 training/train.py --model vgg16
   ```
   Then I get the following error msg:
   
   ```
   File "training/train.py", line 437, in <module>
       args.onnx_model_path, args.data, sgd, args.graph, args.verbosity)
     File "training/train.py", line 295, in run
       model.compile([tx], is_train=True, use_graph=graph, sequential=sequential)
     File "/home/extend/lijiansong/work-space/anaconda2/envs/intel-caffe/lib/python3.6/site-packages/singa/model.py",
line 177, in compile
       self.forward(*inputs)
     File "/home/extend/lijiansong/work-space/anaconda2/envs/intel-caffe/lib/python3.6/site-packages/singa/layer.py",
line 63, in wrapper
       return func(self, *args, **kwargs)
     File "training/train.py", line 191, in forward
       y = self.linear(y)
     File "/home/extend/lijiansong/work-space/anaconda2/envs/intel-caffe/lib/python3.6/site-packages/singa/layer.py",
line 110, in __call__
       return self.forward(*args, **kwargs)
     File "/home/extend/lijiansong/work-space/anaconda2/envs/intel-caffe/lib/python3.6/site-packages/singa/layer.py",
line 61, in wrapper
       self.initialize(*args, **kwargs)
     File "/home/extend/lijiansong/work-space/anaconda2/envs/intel-caffe/lib/python3.6/site-packages/singa/layer.py",
line 45, in wrapper
       'initialize function expects PlaceHolders or Tensors')
   AssertionError: initialize function expects PlaceHolders or Tensors
   ```
   
   Something maybe wrong with the layer initialization?
   
   singa version: 3100(the latest build from the source code of master branch)
   Python version: 3.5.2
   ONNX version: 1.5.0
   
   


----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

For queries about this service, please contact Infrastructure at:
users@infra.apache.org



Mime
View raw message