-
Notifications
You must be signed in to change notification settings - Fork 229
Open
Description
I try to run the py/run_convnet.py in the following way:
- In the path py/
- Add "sys.path.insert('.../convnet')" in the run_convnet.py before ''import convnet as cn"
- Then run "python run_convnet.py ../examples/imagenet/CLS_net_20140621074703.pbtxt ../examples/imagenet/CLS_net_20140621074703.h5 ../examples/imagenet/pixel_mean.h5"
The outputs as follows:
Using board 0
[u'input', u'hidden1_conv', u'hidden1_maxpool', u'hidden1_rnorm', u'hidden2_conv', u'hidden2_maxpool', u'hidden2_rnorm', u'hidden3_conv', u'hidden4_conv', u'hidden5_conv', u'hidden5_maxpool', u'hidden6', u'hidden7', u'output']
python: cudamat_conv_gemm.cu:477: void convUpGemm(cudamat, cudamat_, cudamat*, Shape4D, Shape4D, Shape4D, ConvDesc, float, float, bool): Assertion `output_channel_end - output_channel_begin == num_output_channels3' failed.
Aborted (core dumped)
How can i fix it ? Thank you!
Metadata
Metadata
Assignees
Labels
No labels