히비스서커스의 블로그

pytorch로 모델을 직접 설계하여 모델을 학습시킨 후 model.predict()을 하였으나 model에는 predict 특성이 없다는 에러가 발생하였다. 상황 설계한 모델형식 (간단하게 줄여서 나타냄) import torch import torch.nn as nn class simple_model(nn.Module): def __int__(self, inchannels, out_channels, kernel_size, stride, padding): super().__init__() self.conv = nn.Conv2d(in_channels=in_channels, out_channels=out_channels, kernel_size=kernel_size, stride=stride, padding=..