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
60 Dec 31, 2022
23 Dec 21, 2022
75 Dec 15, 2022
87 Dec 21, 2022
86 Dec 02, 2022
19 Dec 12, 2022
1 Nov 24, 2021
47 Dec 09, 2022
4.5k Dec 31, 2022
10 Dec 29, 2022
1.1k Dec 12, 2022
31 Dec 08, 2022
3 Nov 23, 2022
81 Dec 09, 2022
5 May 22, 2022
97 Dec 23, 2022
44 Dec 31, 2022
166 Dec 11, 2022
3 Aug 17, 2022
215 Dec 30, 2022