Can't get attribute 'ResNet' on <module '__main__'>

Custom으로 만든 Resnet 모델을 저장한 후

다른 파일에서 불러와 쓰려고 하니 발생한 문제

원인

저장된 모델이 built-in ResNet을 쓰지 않아서, 변형된 레즈넷을 Transfer Learning 하는 파일에서 재정의할 필요가 있었음.

해결

전이학습할 파일에 바뀐 레즈넷 코드를 옮겨놓음

conv1x1 = resnet.conv1x1
Bottleneck = resnet.Bottleneck
BasicBlock = resnet.BasicBlock

class ResNet(nn.Module):
  def __init__(self, block, layers, num_classes=3, zero_init_residual=True):
    #---ADDED BY ME ENDED-----#
    #---ADDED BY ME-----#