Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Feat: support no_grad decorator #5947

Closed
wants to merge 4 commits into from
Closed

Feat: support no_grad decorator #5947

wants to merge 4 commits into from

Conversation

wyg1997
Copy link
Contributor

@wyg1997 wyg1997 commented Aug 18, 2021 •
edited
Loading

实现三种改变 grad_mode 的方式，并支持 with ... 和装饰器两种方式：

  • flow.inference_mode(bool mode)
  • flow.grad_enable()
  • flow.no_grad()

以及一个查询当前 grad_mode 的接口：flow.autograd.grad_mode()

image
image
image

@wyg1997
Copy link
Contributor Author

wyg1997 commented Aug 19, 2021

加了新的功能，旧的分支名不再适用，已提新的 PR：#5956

@wyg1997 wyg1997 closed this Aug 19, 2021
@wyg1997 wyg1997 deleted the feat-no_grad_decorator branch September 8, 2021 06:40
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@Flowingsun007 Flowingsun007 Flowingsun007 left review comments

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@wyg1997 @Flowingsun007