galaxy.main

Script to train and evaluate models with optional segmentation.

Module Contents

galaxy.main.parent_dir = b'.'

galaxy.main.all_models

galaxy.main.all_optimizers

galaxy.main.available_datasets = ['WISE', 'ACT']

galaxy.main.parser

galaxy.main.args

galaxy.main.selected_models

galaxy.main.num_epochs

galaxy.main.lr = 0.0001

galaxy.main.momentum

galaxy.main.optimizer_name

galaxy.main.dataset

galaxy.main.device

galaxy.main.train_loader

galaxy.main.val_loader

galaxy.main.test_loader

galaxy.main.experiment

galaxy.main.criterion

galaxy.main.results

galaxy.main.val_results

galaxy.main.model