Building and Training Deep Learning Models in PyTorch
YouTube Viewers YouTube Viewers
53.8K subscribers
1,651 views
0

 Published On Nov 16, 2023

BCS Computational Tutorial Series with Valmiki Kothare, MIT.
In this tutorial, we will use deep learning on EEG and EMG mice data to predict sleep stages (Wakefulness, REM, Non-REM). We will walk through an example Jupyter Notebook in which we load a dataset, preprocess it, build a "residual-attention" network, train our model, and validate our performance on withheld data. In the process of going through the notebook, we will discuss briefly how to run this on OpenMind and how to parallelize training across multiple GPUs, as well as the reasoning behind the network architecture choice and the basic theory of the attention/transformer layer.
Google Colab Notebook - https://colab.research.google.com/dri...

show more

Share/Embed