EquivariantSelfAttention
An implementation of Equivariant e2 convolutional kernals into a convolutional self attention network, applied to radio astronomy data.
This work extends and builds on previous work:
- FR Classification using e2cnn, whose code is here.
- Attention for FR Classificaion, whose code is here. This work makes extensive use of the
e2cnnpackage: e2cnn.
Imports for General Use
import os
import torch
import e2cnn
import pandas as pd
import numpy as np
import configparser as ConfigParser
import utils
Loading a network:
To load a network import the network files, read in a config file, and load the network with the config file parameters:
from networks import AGRadGalNet, VanillaLeNet, testNet, DNSteerableLeNet, DNSteerableAGRadGalNet
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
config = ConfigParser.ConfigParser(allow_no_value=True)
config_path = "configs/CONFIG_NAME"
config.read(config_path)
net = locals()[config['model']['base']](**config['model']).to(device)
Alternatively, if you know which architecture you want to use, only import that and load in the config parameters (i.e. without using locals()):
from networks import DNSteerableAGRadGalNet
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
config = ConfigParser.ConfigParser(allow_no_value=True)
config_path = "configs/CONFIG_NAME"
config.read(config_path)
net = DNSteerableAGRadGalNet(**config['model']).to(device)
Loading a model:
To load the best performing model (according to early stopping) from a given model training session:
path_supliment = config['data']['augment']+'/'
model = utils.utils.load_model(config, load_model='best', device=device, path_supliment=path_supliment)
Testing a model:
Follow the example of evaluation.ipynb for model evaluation.
Defining a model:
To define your own model simply create a new config which follows the examples provided in configs.
Training a model:
To train a model use the train.py using your config file:
python3.8 train.py --config YOUR_CONFIG_NAME.cfg >& logs/YOUR_CONFIG_NAME.log
7 Oct 13, 2022
1 Jan 30, 2022
22 Jun 14, 2022
62 Dec 05, 2022
16 Jul 11, 2022
8 Nov 24, 2022
100 Dec 19, 2022
21 Dec 21, 2022
65 Dec 27, 2022
82 Jan 04, 2023
6 Nov 24, 2022
27 Dec 13, 2022
31 Dec 24, 2022
4 Jul 16, 2022
6.4k Jan 05, 2023
9 Nov 28, 2022
599 Dec 23, 2022
161 Jan 03, 2023
58 Dec 07, 2022
29 Dec 29, 2022