sadaco.pipelines package

Submodules

sadaco.pipelines.ICBHI module

class sadaco.pipelines.ICBHI.ICBHI_Basic_Trainer(configs)[source]

Bases: sadaco.apis.traintest.trainer_base.BaseTrainer

build_dataset()[source]
train_epoch(epoch)[source]
validate_epoch(epoch)[source]
class sadaco.pipelines.ICBHI.ICBHI_Contrast_Trainer(configs)[source]

Bases: sadaco.apis.contrastive.trainer_contrastive.ContrastTrainer

build_dataset()[source]
train_epoch(epoch)[source]
validate_epoch(epoch)[source]

sadaco.pipelines.build_modules module

sadaco.pipelines.build_modules.build_optimizer(model, train_configs, trainables=None)[source]
sadaco.pipelines.build_modules.build_dataloader(dataset, train_configs, data_configs)[source]
sadaco.pipelines.build_modules.build_criterion(name, mixup=False, **kwargs)[source]

sadaco.pipelines.scheduler module

class sadaco.pipelines.scheduler.CosineAnnealingWarmUpRestarts(optimizer, T_0, T_mult=1, eta_max=0.1, T_up=0, gamma=1.0, last_epoch=- 1, verbose=False)[source]

Bases: torch.optim.lr_scheduler._LRScheduler

get_lr()[source]
step(epoch=None)[source]
class sadaco.pipelines.scheduler.BaseScheduler(train_configs, optimizer: torch.nn.modules.module.Module, model: torch.nn.modules.module.Module, exp_id=None, parallel=None)[source]

Bases: object

step(train_stats, valid_stats, *lr_sched_args)[source]

Module contents