Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
Asthestarsfalll committed Feb 20, 2023
1 parent b181af6 commit 6329d1b
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
# flip paddle model generator
#
import numpy as np
# from save_model import saveModel
from save_model import saveModel
import paddle
import sys

Expand All @@ -11,7 +11,7 @@ def flip(name: str, x, axis):
paddle.enable_static()

with paddle.static.program_guard(paddle.static.Program(), paddle.static.Program()):
datpaddle.static.data(name='x', shape=x.shape, dtype=x.dtype)
data = paddle.static.data(name='x', shape=x.shape, dtype=x.dtype)
out = paddle.flip(data, axis=axis)

cpu = paddle.static.cpu_places(1)
Expand Down

0 comments on commit 6329d1b

Please sign in to comment.