-
Notifications
You must be signed in to change notification settings - Fork 157
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
MobilenetV1 profile for optimizing #308
Comments
test depthwise_conv op: input dim:[32,32,224,224]
|
import paddle.fluid as fluid data = fluid.data(name='input', shape = [32, 32, 224, 224], dtype='float32') result = fluid.layers.conv2d(input=data, place = fluid.CUDAPlace(0) exe.run(fluid.default_startup_program()) in_data1 = np.random.rand(32, 32, 224, 224).astype('float32') all_time = 0 fluid.profiler.start_profiler('All') start = time.time() if i >=5: fluid.profiler.stop_profiler('total') tf: data = tf.placeholder(name='input', shape=[32, 32, 224, 224], dtype=tf.float32) in_data = np.random.rand(32, 32, 224, 224).astype('float32') with tf.Session() as sess: print('argsort time:', all_time / 5.0) |
CUDNN 7.5.1
develop cuda
tensorflow
|
Paddle和TensorFlow的API性能测试和profile框架可依据benchmark/api/paddle/abs.py和api/tensorflow/abs.py编写,并往benchmark里面提交测试脚本。 |
优化后
|
Profile MobilenetV1 to optimize depthwise conv.
The text was updated successfully, but these errors were encountered: