반응형 [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 다음 반응형
