Method __init__

core/mengine.py:32–42

(self, local_rank, world_size=0, DDP=False, SyncBatchNorm=False)

class TrainEngine(object):
    def __init__(self, local_rank, world_size=0, DDP=False, SyncBatchNorm=False):
        # init setting
        self.local_rank = local_rank
        self.world_size = world_size
        self.device_ = f'cuda:{local_rank}'
        # create tool
        self.cls_meter_ = MultilabelClassificationMetric()
        self.loss_meter_ = MultiClassificationMetric()
        self.top1_meter_ = MultiClassificationMetric()
        self.DDP = DDP
        self.SyncBN = SyncBatchNorm

    def create_env(self, cfg):
        # create network