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

fix pre-commit

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