728x90
반응형

pytorch 2

[PyTorch] MLP 모델 만들기 2편 - 학습 모델 구축하기

지난 1편에서는 학습 데이터셋 만들어 보았습니다.https://notesfromyoungdeveloper.tistory.com/25 [PyTorch] MLP 모델 만들기 1편 - 학습 데이터셋 만들기PyTorch에서 MLP 모델을 만드는 방법에 대해 알아보겠습니다.MLP는 Multi-Layer Perceptron의 약자로, 입력층(input layer)과 출력층(output layer) 사이에 1개 이상의 은닉층(hidden layer)이 있는 구조의 퍼셉트론을notesfromyoungdeveloper.tistory.com 2편에서는 MLP 모델 구축하는 방법에 대해 알아보겠습니다.먼저, 전체 코드는 아래와 같고, 하나씩 살펴보겠습니다.class XORModel(pl.LightningModule): ..

PyTorch Lightning 설치 방법

PyTorch Lightning은 딥러닝 프레임워크로, 이를 활용하면 누구나 쉽게 딥러닝 모델을 학습할 수 있습니다.TensorFlow에 Keras가 있다면, PyTorch에는 PyTorch Lightning이 있다고 봐주시면 될 것 같습니다. PyTorch Lightning은 아래 명령어로 간단하게 설치할 수 있습니다.pip install pytorch-lightning 설치가 완료되면, 아래와 같이 import 해서 사용할 수 있습니다.간단하게 버전 정보를 출력해 보도록 하겠습니다.import pytorch_lightning as plprint(f'PyTorch lightning version: {pl.__version__}') 위 출력 결과를 통해, PyTorch가 잘 설치되었는지, 어느 버전이 설..

728x90
반응형