Non diff layers #1498
Replies: 1 comment 4 replies
-
|
You can use Module::no_grad on the layers you don't want to optimize. |
Beta Was this translation helpful? Give feedback.
-
|
Is there an example? |
Beta Was this translation helpful? Give feedback.
-
|
I don't think we do at the moment. |
Beta Was this translation helpful? Give feedback.
-
|
ok,thanks maybe I will write one myself to get a better understanding of the framework |
Beta Was this translation helpful? Give feedback.
-
|
That would be really appreciated! |
Beta Was this translation helpful? Give feedback.
-
Can I make the optimizer skip custom layers(non-diff layers)?
Beta Was this translation helpful? Give feedback.
All reactions