The model architecture consists of two parts: the generative adversarial network (implemented in Pytorch) and the forward solver (MRST). The coupling between the two is implemented in ...