def optimizers(): """ print overview of all optimizers """ print('\n', 'Optimizers:') from torch.optim import Optimizer as O optimizers = all_subclasses(O) print_options(optimizers)
def losses(): """ print overview of all losses """ print('\n', 'Losses:') from hibashi.losses.loss import Loss as L losses = all_subclasses(L) print_options(losses)
def datasets(): """ print overview of all datasets """ print('\n', 'Datasets:') from hibashi.data.datasets.datasets import Dataset as D datasets = all_subclasses(D) print_options(datasets)
def networks(): """ print overview of all networks """ print('\n', 'Networks:') from hibashi.networks.network import Network as N networks = all_subclasses(N) print_options(networks)
def models(): """ print overview of all models """ print('\n', 'Models:') from hibashi.models import Model as M models = all_subclasses(M) print_options(models)
def all_models(cls): subclasses = all_subclasses(cls) models = {cls.__name__.lower(): cls for cls in subclasses} return models