occamypy icon indicating copy to clipboard operation
occamypy copied to clipboard

Add torch.nn.Module operator wrapper

Open fpicetti opened this issue 3 years ago • 1 comments

CNNs can be seen as nonlinear operators.

Two ideas here:

  1. use a CNNs as operator where only the forward is implemented; CNN can be used as regularizers (link)
  2. implement a VectorTorch method to build a torch.nn.Parameter object, to be passed to torch.optim.

fpicetti avatar Mar 03 '21 14:03 fpicetti