Google Drive1 [Pytorch] 구글 드라이브에서 pretrained 모델(pth) torchvision 다운폴더로 받아서 실행하기. 안녕하세요 pulluper 입니다. 최근에 개발을 하다가 제가 학습시켜놓은 pth 파일을 torchvision처럼 다운받아서 바로 돌리도록 코드를 작성했습니다. import os import torch import gdown def download_pretrained_model(pth_name, file_id=None): google_path = 'https://drive.google.com/uc?id=' if file_id is None: return None torch_dir = torch.hub.get_dir() output_name = pth_name if os.path.exists(os.path.join(torch_dir, 'checkpoints', output_name)): print("Alr.. 2023. 5. 23. 이전 1 다음
