occamypy
occamypy copied to clipboard
Add torch.nn.Module operator wrapper
CNNs can be seen as nonlinear operators.
Two ideas here:
- use a CNNs as operator where only the forward is implemented; CNN can be used as regularizers (link)
- implement a
VectorTorch
method to build atorch.nn.Parameter
object, to be passed totorch.optim
.