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