Non diff layers

[DanielMadmon]

Can I make the optimizer skip custom layers(non-diff layers)?

[nathanielsimard]

You can use Module::no_grad on the layers you don't want to optimize.

[DanielMadmon]

Is there an example?

[nathanielsimard]

I don't think we do at the moment.

[DanielMadmon]

ok,thanks maybe I will write one myself to get a better understanding of the framework

[nathanielsimard]

That would be really appreciated!