From 2fc4e8c3ad512528b941c85358f47174e7eeaebb Mon Sep 17 00:00:00 2001
From: TomorrowIsAnOtherDay <zhoubo01@baidu.com>
Date: Tue, 13 Nov 2018 15:18:19 +0800
Subject: [PATCH] support for Paddle-v1.1 (#24)

---
 Dockerfile                    | 2 +-
 parl/layers/layer_wrappers.py | 4 ----
 2 files changed, 1 insertion(+), 5 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index f89aa20..8e4e7a6 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -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
diff --git a/parl/layers/layer_wrappers.py b/parl/layers/layer_wrappers.py
index db293a6..fcafe61 100644
--- a/parl/layers/layer_wrappers.py
+++ b/parl/layers/layer_wrappers.py
@@ -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_()
-- 
GitLab