提交 95de3b44 编写于 作者: H Helin Wang

fix pre-commit

上级 1561ba59
...@@ -5,6 +5,7 @@ import paddle.v2 as paddle ...@@ -5,6 +5,7 @@ import paddle.v2 as paddle
with_gpu = os.getenv('WITH_GPU', '0') != '0' with_gpu = os.getenv('WITH_GPU', '0') != '0'
def softmax_regression(img): def softmax_regression(img):
predict = paddle.layer.fc( predict = paddle.layer.fc(
input=img, size=10, act=paddle.activation.Softmax()) input=img, size=10, act=paddle.activation.Softmax())
......
...@@ -21,6 +21,7 @@ from resnet import resnet_cifar10 ...@@ -21,6 +21,7 @@ from resnet import resnet_cifar10
with_gpu = os.getenv('WITH_GPU', '0') != '0' with_gpu = os.getenv('WITH_GPU', '0') != '0'
def main(): def main():
datadim = 3 * 32 * 32 datadim = 3 * 32 * 32
classdim = 10 classdim = 10
......
...@@ -5,6 +5,7 @@ import os ...@@ -5,6 +5,7 @@ import os
with_gpu = os.getenv('WITH_GPU', '0') != '0' with_gpu = os.getenv('WITH_GPU', '0') != '0'
def get_usr_combined_features(): def get_usr_combined_features():
uid = paddle.layer.data( uid = paddle.layer.data(
name='user_id', name='user_id',
......
...@@ -17,6 +17,7 @@ import paddle.v2 as paddle ...@@ -17,6 +17,7 @@ import paddle.v2 as paddle
with_gpu = os.getenv('WITH_GPU', '0') != '0' with_gpu = os.getenv('WITH_GPU', '0') != '0'
def convolution_net(input_dim, class_dim=2, emb_dim=128, hid_dim=128): def convolution_net(input_dim, class_dim=2, emb_dim=128, hid_dim=128):
data = paddle.layer.data("word", data = paddle.layer.data("word",
paddle.data_type.integer_value_sequence(input_dim)) paddle.data_type.integer_value_sequence(input_dim))
......
...@@ -4,6 +4,7 @@ import paddle.v2 as paddle ...@@ -4,6 +4,7 @@ import paddle.v2 as paddle
with_gpu = os.getenv('WITH_GPU', '0') != '0' with_gpu = os.getenv('WITH_GPU', '0') != '0'
def save_model(parameters, save_path): def save_model(parameters, save_path):
with open(save_path, 'w') as f: with open(save_path, 'w') as f:
parameters.to_tar(f) parameters.to_tar(f)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册