그냥 편하게 한국말로 할게요 ^^… 굳이 영어로 하라는 법은 없어서…
보니까 model.children 사용하셨는데, 이렇게 하시면 sequential 한 모델, (vgg같은 residual connection이나 다양한 operation이 없는 모델) 에서만 그렇게 가능하고 나머지에서는 불가능 해요
그래서 forward hook을 사용하셔야해요
  
      

      velog.io
  

  
    

Introduction to torch.autograd 2

  Week7(CV), Day 33. Introduction to torch.autograd


  

  
    
    
  

  


여기에 친절하게 hook function 사용하는 방법, 어떻게 동작하는지까지 알려주니까 이 이상의 설명은 생략할게요
제가 코딩까지 해드릴정도로 시간이 많지는 않아서 ^^;;
에러를 보니 제가 말씀드린 부분이 맞을 것 같아요. for children 사용하지 마시구요 hook function 사용해보세요!
정말 다양한 코딩이 가능해지고, 더 고급 프로그래머가 되실꺼예요!