ChengYue's Cat

Trainer

Trainer类的定义

class Trainer(DefaultTrainer):
    """
    继承自DefaultTrainer
    """
    
    @classmethod
    def build_evaluator(cls, cfg, dataset_name, output_folder=None):
        pass
    
    @classmethod
    def build_train_loader(cls, cfg):
        pass
    
    @classmethod
    def build_test_loader(cls, cfg):
        pass
    
    def build_writers(self):
        # 暂时还不清楚具体的功能作用，大概是记录训练过程产生的结果数据什么的
        pass
    
    @classmethod
    def build_lr_scheduler(cls, cfg, optimizer):
        # 暂时不清楚
        pass
    
    @classmethod
    def build_optimizer(cls, cfg, model):
        # 构造一个优化器
        pass
    
    @classmethod
    def test_with_TTA(cls, cfg, model):
        # 作用暂时不清楚
        pass

This site is open source. Improve this page.