提交 2fc4e8c3 编写于 作者: T TomorrowIsAnOtherDay 提交者: Hongsheng Zeng

support for Paddle-v1.1 (#24)

上级 b32f8940
...@@ -15,7 +15,7 @@ ...@@ -15,7 +15,7 @@
# A dev image based on paddle production image # A dev image based on paddle production image
FROM paddlepaddle/paddle:latest-gpu FROM paddlepaddle/paddle:1.1.0-gpu-cuda9.0-cudnn7
RUN apt-get install -y cmake RUN apt-get install -y cmake
RUN pip install -i https://pypi.tuna.tsinghua.edu.cn/simple gym RUN pip install -i https://pypi.tuna.tsinghua.edu.cn/simple gym
......
...@@ -179,7 +179,6 @@ def fc(size, ...@@ -179,7 +179,6 @@ def fc(size,
num_flatten_dims=1, num_flatten_dims=1,
param_attr=None, param_attr=None,
bias_attr=None, bias_attr=None,
use_mkldnn=False,
act=None, act=None,
name=None): name=None):
""" """
...@@ -200,7 +199,6 @@ def fc(size, ...@@ -200,7 +199,6 @@ def fc(size,
num_flatten_dims=num_flatten_dims, num_flatten_dims=num_flatten_dims,
param_attr=self.param_attr, param_attr=self.param_attr,
bias_attr=self.bias_attr, bias_attr=self.bias_attr,
use_mkldnn=use_mkldnn,
act=act, act=act,
is_test=is_test) is_test=is_test)
...@@ -410,7 +408,6 @@ def conv2d(num_filters, ...@@ -410,7 +408,6 @@ def conv2d(num_filters,
param_attr=None, param_attr=None,
bias_attr=None, bias_attr=None,
use_cudnn=True, use_cudnn=True,
use_mkldnn=False,
act=None, act=None,
name=None): name=None):
""" """
...@@ -437,7 +434,6 @@ def conv2d(num_filters, ...@@ -437,7 +434,6 @@ def conv2d(num_filters,
param_attr=self.param_attr, param_attr=self.param_attr,
bias_attr=self.bias_attr, bias_attr=self.bias_attr,
use_cudnn=use_cudnn, use_cudnn=use_cudnn,
use_mkldnn=use_mkldnn,
act=act) act=act)
return Conv2D_() return Conv2D_()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册