singa: AssertionError for the ONNX training testcases?

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

About this issue

  • Original URL
  • State: closed
  • Created 4 years ago
  • Comments: 18 (17 by maintainers)

Most upvoted comments

@joddiy: FYR as below. i ran on panda13, when GPU memory was enough for the model: $ cd singa $ python examples/onnx/training/train.py --model resnet152v1 Error: " thao@panda13:/hdd2/thao/singa$ python examples/onnx/training/train.py --model resnet152v1

2020-10-13 20:09:27,800 Downloading https://s3.amazonaws.com/onnx-model-zoo/resnet/resnet152v1/resnet152v1.tar.gz Traceback (most recent call last): File “examples/onnx/training/train.py”, line 352, in <module> args.data, sgd, args.graph, args.verbosity) File “examples/onnx/training/train.py”, line 216, in run model.compile([tx], is_train=True, use_graph=graph, sequential=sequential) File “/hdd2/thao/conda/miniconda3/envs/sing/lib/python3.7/site-packages/singa/model.py”, line 177, in compile self.forward(*inputs) File “/hdd2/thao/conda/miniconda3/envs/sing/lib/python3.7/site-packages/singa/layer.py”, line 61, in wrapper return func(self, *args, **kwargs) File “examples/onnx/training/train.py”, line 119, in forward y = self.linear(y) File “/hdd2/thao/conda/miniconda3/envs/sing/lib/python3.7/site-packages/singa/layer.py”, line 108, in call return self.forward(*args, **kwargs) File “/hdd2/thao/conda/miniconda3/envs/sing/lib/python3.7/site-packages/singa/layer.py”, line 59, in wrapper self.initialize(*args, **kwargs) File “/hdd2/thao/conda/miniconda3/envs/sing/lib/python3.7/site-packages/singa/layer.py”, line 43, in wrapper ‘initialize function expects PlaceHolders or Tensors’) AssertionError: initialize function expects PlaceHolders or Tensors"

env:

  • python 3.7
  • singa 3.1.0.rc1 (conda)
  • singa git - dev branch