--- a +++ b/model/README.md @@ -0,0 +1,6 @@ +# Pre-processing and model definition + +Follow the preprocessing steps from writing/preprocessingoverview.png to generate the required base data + +- data_loader.py contains the data loader used by pytorch +- net.py contains the definition of the neural network, loss functions and metrics