반응형 [MMEngine] Tutorial - Runner Runner - MMEngine안의 integrator로 모든 module 및 framework를 책임짐. - Runner's API documentation Simple Example import torch import torch.nn as nn import torch.nn.functional as F from torch.utils.data import Dataset from mmengine.model import BaseModel from mmengine.evaluator import BaseMetric from mmengine.registry import MODELS, DATASETS, METRICS @MODELS.register_module() class MyAwesomeModel(BaseMode.. 2024. 1. 14. [MMEngine] 빠른 Tutorial Model 빌드 - MMEngine내에서 모델은 BaseModel을 상속 받음 - 입력값으로 forward 메서드는 mode라는 인자를 받아야함. - training에서 mode는 loss, forward 메서드는 loss키를 포함한 dict를 반환 - validation에서 mode는 predict, forward 메서드는 예측값과 label들을 반환 import torch.nn.functional as F import torchvision from mmengine.model import BaseModel class MMResNet50(BaseModel): def __init__(self): super().__init__() self.resnet = torchvision.models.resnet50().. 2024. 1. 14. 이전 1 다음 반응형
