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

support for Paddle-v1.1 (#24)

上级 b32f8940
......@@ -15,7 +15,7 @@
# 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 pip install -i https://pypi.tuna.tsinghua.edu.cn/simple gym
......
......@@ -179,7 +179,6 @@ def fc(size,
num_flatten_dims=1,
param_attr=None,
bias_attr=None,
use_mkldnn=False,
act=None,
name=None):
"""
......@@ -200,7 +199,6 @@ def fc(size,
num_flatten_dims=num_flatten_dims,
param_attr=self.param_attr,
bias_attr=self.bias_attr,
use_mkldnn=use_mkldnn,
act=act,
is_test=is_test)
......@@ -410,7 +408,6 @@ def conv2d(num_filters,
param_attr=None,
bias_attr=None,
use_cudnn=True,
use_mkldnn=False,
act=None,
name=None):
"""
......@@ -437,7 +434,6 @@ def conv2d(num_filters,
param_attr=self.param_attr,
bias_attr=self.bias_attr,
use_cudnn=use_cudnn,
use_mkldnn=use_mkldnn,
act=act)
return Conv2D_()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册