[Colab] 꿀팁 및 주의사항

사실 꿀팁이랑 주의사항이라고 할 것이 없다. 코랩을 쓰고있는 사람들은 이미 알 것 같은 이야기들 뿐이긴하다.

그래도 코랩을 처음 이용하는 사람들에게 도움이 될까하여 기록해둔다.

 

- Setting

Runtime --> Change runtime type --> Hardware accelerator에서 CPU를 GPU를 바꿀 것

학습시키는 경우, CPU와 GPU 차이는 매우 크다... 

 

- GPU 주의 사항

 Colab 무료 버전을 쓰고 있다면 GPU 사용 제한이 있다.

너무 많이 런타임을 지속하고 있으면 끊기고, 한동안 사용하지 못한다.

나같은 경우, 학습시키고 결과물(best.pt, last.pt, 결과물 수치, 테스트 결과 등)을 모두 Google drive로 마운트했다.

 

마운트 코드

# Google Drive를 Colab에 마운트하기
from google.colab import drive
drive.mount('/content/drive')

# Google Drive에 저장할 경로 지정하기
import os
save_path = 'google drive 저장 디렉토리'
os.makedirs(save_path, exist_ok=True)

import locale
locale.getpreferredencoding = lambda: "UTF-8"
!pip install transformers==4.28.1

# 데이터 저장
!cp -r 저장할 데이터 경로 $저장할 경로

# ex
!cp -r /content/runs/detect/train/* $save_path

 

추가 설명

import locale : 인코딩 문제를 예방함.

local.getpreferredencoding... : 인코딩을 utf-8로 한다는 것

transformers :자연어 처리(NLP) 모델을 다루는 데 사용되는 강력한 라이브러리

 

- 런타임 주의사항

 학습시킬 때, 다른 작업을 하기 마련인데, 계속 확인해주어야한다.

중간에 종종 끊기고, 구글 인증을 해야할 때도 있다.

나같은 경우, 런타임 끊김 방지를 해놓는데, 그래도 항상 불안해서 다른 작업하다가도 확인한다.

 

- 런타임 끊김 방지 코드

https://basically.tistory.com/45

function ClickConnect(){
console.log("Working"); 
document.querySelector("colab-toolbar-button").click() 
}setInterval(ClickConnect, 1800000)

 

 

일단 여기에서 글을 마치고 더 발견한 좋은 점이나 주의사항이 있다면 마저 쓰겠다.