W2 PyTorch Basics for AI

soonyear·2023년 11월 13일
0

ai-tech-review

목록 보기
2/3

Colab에 ssh로 접근 방법

  1. colab-ssh 라이브러리 설치
pip install colab-ssh
  1. ngrok(웹사이트)에서 인증 토큰을 받는다
  2. ssh 접근을 개방한다
from colab_ssh import launch_ssh
launsh_ssh(TOKEN, PASSWORD)
  1. ssh로 접근한다.

Dataloader

Dataloader 사용시 주의할 점

for epoch in epochs:
    x, y = next(iter(dataloader))

이런식으로 사용 시 dataloader에 엄청난 병목이 생긴다.
왜냐하면 iter(dataloader)가 호출 될 때마다 새로운 dataloder객체를 정의하게 되는 방식으로 작동하게 되기 때문이다.
올바른 사용법은 다음과 같다.

data_iterator = iter(dataloader)
for epoch in epochs:
    x, y = next(data_iterator)

2023-11-15

Torch Summary

모델의 input 크기를 설정하면 모델의 크기과 intermediate tensor, output tensor의 크기를 확인 할 수 있음.

2023-11-16

Hyperparameter Tuning

보통 Hyperparameter optimization이라고도 하는듯.
Ray Tune이라는 라이브러리를 많이 사용한다고 한다.

PyTorch Practice

아래 코드를 보니 되게 신기한 것을 배웠다.
total_loss += iter_loss를 하게되면 total_loss에 computation graph with one AddBackward function node가 계속 추가된다고 한다.

total_loss = 0

for x in range(10):
  # assume loss is computed 
  iter_loss = torch.randn(3,4).mean()
  iter_loss.requires_grad = True     # losses are supposed to differentiable
  total_loss += iter_loss            # use total_loss += iter_loss.item() instead
profile
A.I. Developer / Engineer / Researcher

0개의 댓글