목록layer 설정 (1)

다시 이음

Pytorch 모델 생성 및 layer 설정

안녕하세요. 오늘은 Pytorch에서 모델을 생성하고 배정하는 방법에 대해서 알아보겠습니다. 모델을 정의하는 방법에서 tensorflow와 다른 점이 보여서 그 점을 비교하면서 하면 좋을 것 같습니다. Pytorch - Module 클래스 Model 을 정의할 때에는 Module 클래스를 무조건 상속 받아야합니다. forward() 메소드는 모든 자식클래스에서 반드시 오버라이딩 해야합니다. 공식문서 // https://pytorch.org/docs/stable/generated/torch.nn.Module.html#torch.nn.Module 공식 문서를 통해 메소드를 보면 좋습니다.(eval(), train(), parameters(), state_dict(), to() 등등) class Model(n..
Pre_Onboarding by Wanted(자연어 처리) 2022. 3. 2. 18:26
이전 Prev 1 Next 다음