possibly edit the network architecture with background knowledge on primal dual and what convolutions etc. might be good to do and do that training