[WEEK02] Tips

▮ Warning이 안뜨게 하는 방법(사용하기를 권장)

import warnings
warnings.filterwarnings("ignore")

 

▮ 디렉토리가 존재하지 않으면 만들기

import os

if not os.path.exists(MODEL_PATH):
	os.makedirs(MODEL_PATH)

import os

os.makedirs(MODEL_PATH, exist_ok=True)

 

▮ 안읽히는 파일은 remove를 시켜주면 좋다.

f_path = []
for (dirpath, dirnames, filenames) in walk(mypath):
    f_path.extend([os.path.join(dirpath, filename) for filename in filenames])

from PIL import Image

for f_name in f_path:
    try:
        Image.open(f_name)
    except Exception as e:
        print(e)
        os.remove(f_name)

 

▮ 아래 코드처럼 my_model에 모델을 불러오고 다시 device로 부르는 작업을 해주는게 좋다.

my_model = MyNewNet()
my_model = my_model.to(device)

 

▮ model.eval() → model._eval()을 사용하자

 

▮ 재현성을 위한 SEED 설정

# Set random seed
SEED = 2022
np.random.seed(SEED)
torch.manual_seed(SEED)
torch.cuda.manual_seed(SEED)  # type: ignore
torch.backends.cudnn.deterministic = True  # type: ignore
torch.backends.cudnn.benchmark = True  # type: ignore