Exercise 3

The goal of this exercise is to build a neural network that predicts lid driven cavity flow solutions based on varying Reynolds numbers using pytorch. You will implement the data generation, preparation, model architecture, training, and evaluation.