Machine Learning in Image Processing

Week 02 - Neural Net from Scratch


The second week will build a neural network with one dense layer but using only tensors and standard operations. Then we update the definition using the PyTorch library to offer better functionality.


Goals:

  • understanding the neural network representation and the optimization process,
  • using `Dataset` and `DataLoader` classes to manipulate training data,
  • learning the basics of the `nn.Module` class and `torch.optim` library,
  • switching between 'GPU' and 'CPU' computation.