Implement a custom function inside the model

vision
3

Thanks a lot!
In this case, shall I implement backwards method, how described here? Defining backward() function in nn.module?

1 Like