Skip to content

Latest commit

 

History

History
17 lines (8 loc) · 817 Bytes

File metadata and controls

17 lines (8 loc) · 817 Bytes

Residual-Attention-Aware-Network

A ResNet that uses attention modules to classify the images in the CIFAR-10 dataset

Link to the paper: https://arxiv.org/abs/1704.06904

Link to the dataset: https://www.cs.toronto.edu/~kriz/cifar.html

The dataset can also be loaded through the keras library.

You can run the notebook on Google Colab or your local machine. In case of Colab, please change the model directory path to your preferred directory path on your Google Drive. In case of a local machine, please save the model directly to a folder on your machine, instead of using the drive library.

The Resnet Attention Aware Network architecture

The Resnet Attention Aware Network architecture