pytorch
-
import torch.nn as nn 딥러닝을 배우면서 pytorch를 쓰기 시작하면 다음과 같은 코드를 많이 봤을 것이다. torch.nn은 neural network (인공 신경망) 에 사용되는 다양한 클래스와 모듈이 포함되어있다. 예를 들어 위 코드에서도 보면 torch.nn을 import 하고 Linear, ReLU 등을 사용하는 것을 볼 수 있다. https://pytorch.org/docs/stable/nn.html torch.nn — PyTorch 2.0 documentation Shortcuts pytorch.org 파이토치 공식 문서를 보면 딥러닝에서 사용되는 다양한 layer, container 등을 확인 할 수 있다.
[Python] import torch.nn as nn 역할import torch.nn as nn 딥러닝을 배우면서 pytorch를 쓰기 시작하면 다음과 같은 코드를 많이 봤을 것이다. torch.nn은 neural network (인공 신경망) 에 사용되는 다양한 클래스와 모듈이 포함되어있다. 예를 들어 위 코드에서도 보면 torch.nn을 import 하고 Linear, ReLU 등을 사용하는 것을 볼 수 있다. https://pytorch.org/docs/stable/nn.html torch.nn — PyTorch 2.0 documentation Shortcuts pytorch.org 파이토치 공식 문서를 보면 딥러닝에서 사용되는 다양한 layer, container 등을 확인 할 수 있다.
2023.08.14 -
pytorch를 사용하다 보면 torch.no_grad()라는 걸 종종 확인할 수 있다. torch.no_grad()의 역할은 딥러닝 모델을 test, 즉 inference를 할 때에는 backpropagation이 진행되지 않는다. 즉, gradient 연산을 전혀 사용하지 않는다. 그렇기 때문에 굳이 torch에서 auto gradient 연산이 돌아가게 켜 둘 필요가 없다. 따라서 위 pytorch 공식 문서에 나와있는 것 처럼 gradient 연산을 꺼서 불필요한 메모리 사용량을 줄이는 것이 목적이다. https://pytorch.org/docs/stable/generated/torch.no_grad.html no_grad — PyTorch 2.0 documentation Shortcuts pyt..
[Python] torch.no_grad() 역할pytorch를 사용하다 보면 torch.no_grad()라는 걸 종종 확인할 수 있다. torch.no_grad()의 역할은 딥러닝 모델을 test, 즉 inference를 할 때에는 backpropagation이 진행되지 않는다. 즉, gradient 연산을 전혀 사용하지 않는다. 그렇기 때문에 굳이 torch에서 auto gradient 연산이 돌아가게 켜 둘 필요가 없다. 따라서 위 pytorch 공식 문서에 나와있는 것 처럼 gradient 연산을 꺼서 불필요한 메모리 사용량을 줄이는 것이 목적이다. https://pytorch.org/docs/stable/generated/torch.no_grad.html no_grad — PyTorch 2.0 documentation Shortcuts pyt..
2023.08.14 -
오류 코드 RuntimeError: PytorchStreamReader failed reading zip archive: failed finding central directory semgentation 모델을 mmdet에서 돌릴 때 발생한 에러 코드이다. 해당 오류는 학습한 weight 파일이 깨졌을 경우 발생한다. 해결 방법 학습된 weight 파일들이 여러개 있다면 다른 weight 파일을 가지고 inference를 돌려본다.
RuntimeError: PytorchStreamReader failed reading zip archive: failed finding central directory오류 코드 RuntimeError: PytorchStreamReader failed reading zip archive: failed finding central directory semgentation 모델을 mmdet에서 돌릴 때 발생한 에러 코드이다. 해당 오류는 학습한 weight 파일이 깨졌을 경우 발생한다. 해결 방법 학습된 weight 파일들이 여러개 있다면 다른 weight 파일을 가지고 inference를 돌려본다.
2023.08.11