반응형 컨테이너1 [파이토치] 03 - 모델을 만드는 여러가지 방법 모듈을 만드는 법¶ 파이토치 (라이트닝 포함) 신경망의 뼈대라고 할 수 있는 모듈을 만드는 방법은 여러가지가 있는데, 그중에서 대표적인 몇가지 정리하면 In [7]: import pytorch_lightning as pl from torchinfo import summary Lighting 의 모듈 컨테이너를 사용하는 방법¶ 가장 기본적으로 사용되는 방법으로 init 에서 필요한 layer들을 정의하고, 그 layer 들이 어떻게 서로 연결되는지를 forward 에서 정의하는 방법 In [12]: import torch.nn as nn import torch.nn.functional as F class Model( pl.LightningModule ): def __init__(self): super()... 2023. 10. 21. 이전 1 다음 반응형