ssds.pipeline¶

ssds.pipeline.pipeline_anchor_apex¶

class ssds.pipeline.pipeline_anchor_apex.ModelWithLossBasic(model, cls_criterion, loc_criterion, num_classes, match, center_sampling_radius)[source]¶

Bases: torch.nn.modules.module.Module

Class use to help the gpu memory becomes more balance in ddp model

ssds.pipeline.pipeline_anchor_apex.train_anchor_based_epoch(model, data_loader, optimizer, anchors, writer, epoch, device, local_rank)[source]¶: the pipeline for training

ssds.pipeline.pipeline_anchor_basic¶

ssds.pipeline.pipeline_anchor_basic.eval_anchor_based_epoch(model, data_loader, decoder, cls_criterion, loc_criterion, anchors, num_classes, writer, epoch, device)[source]¶: the pipeline for evaluation

ssds.pipeline.pipeline_anchor_basic.train_anchor_based_epoch(model, data_loader, optimizer, cls_criterion, loc_criterion, anchors, num_classes, match, center_sampling_radius, writer, epoch, device)[source]¶: the pipeline for training