728x90
전체 글
544

[Pytorch] Optimizer learning rate 가져오기

다음과 같이 lr_shceduler에서 lr을 가져올 수 있다. optimizer.step() lr_scheduler.step() # get learning rate lr = lr_scheduler.get_last_lr() # or lr = lr_scheduler.optimizer.param_groups[0]['lr'] 하위 버전에서는 lr_scheduler.get_lr() 도 실행되지만 상위 버전에서는 get_lr()이 deprecated될거라고 warning이 뜬다. [ref] https://stackoverflow.com/questions/52660985/pytorch-how-to-get-learning-rate-during-training/52671057

Temp 2021.11.27

[Python] 파일 or 폴더 복사

import shutil txt_path = './class_train/txt' train_path = './class_train/train' dir_list = os.listdir(txt_path) for _dir in dir_list: txt_list = os.listdir(os.path.join(txt_path, _dir)) for txt in txt_list: shutil.copy(os.path.join(txt_path, _dir, txt), os.path.join(train_path, txt)) shutil import shutil.copy를 사용하여 파일 복사 위 예시 코드같은 경우 여러 폴더 안에 파일이 들어가있는 것을 하나의 폴더로 합칠때 사용하였다.

Temp 2021.11.25
728x90