pinpom commented on issue #807: URL: https://github.com/apache/singa/issues/807#issuecomment-707701035
@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 ---------------------------------------------------------------- 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: [email protected]
