-
Notifications
You must be signed in to change notification settings - Fork 5.7k
New issue
Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? # to your account
[Dy2St] transforms.RandomErasing Support static mode #49617
Conversation
你的PR提交成功,感谢你对开源项目的贡献! |
top = paddle.static.nn.cond( | ||
erase_h < h and erase_w < w, | ||
lambda: paddle.uniform( | ||
shape=[1], min=0, max=h - erase_h + 1 | ||
).astype("int32"), | ||
lambda: zero, | ||
) | ||
|
||
left = paddle.static.nn.cond( | ||
erase_h < h and erase_w < w, | ||
lambda: paddle.uniform( | ||
shape=[1], min=0, max=w - erase_w + 1 | ||
).astype("int32"), | ||
lambda: zero, | ||
) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
这两行不用 randint
而使用 uniform
,是因为 randint
的 high
low
参数似乎不支持静态图 Variable
这是 issue 地址: #49620
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
PR types
New features
PR changes
OPs
Describe