전체 글 (31) 썸네일형 리스트형 timm 사용가능한 pretrained model weights 리스트 timm (PyTorch Image Models) 에서는 사전 학습된 model weights를 쉽게 다운받아서 활용할 수 있다. 이때 어떤 Pretrained model weights가 사용가능하고, 원하는 Weights를 사용하기 위해서 어떤 keyword를 사용해야 하는지를 간단하게 적어본다. 먼저, 사용가능한 Pretrained model weights 리스트를 뽑기 위해서는 아래와 같이 입력해주면 된다. import timm print(timm.list_models(pretrained=True)) 모델은 아래와 같이 생성할 수 있다. model = timm.create_model('efficientnet_b0.ra_in1k', pretrained=True) 자세한 내용은 공식 Documentat.. Distributed learning이 시작되지 않는 경우 Tensorflow docker를 새로 생긴 A100 GPU 서버에 옮겨놓고 코드를 돌려보려고 하는데트레이닝이 시작이 안되고... 계속 멈춰 있는 현상이 발생하였다.(강제 종료도 안되고... ㅠㅠ) Tensorflow에서 다음과 같이 Mirrored strategy를 사용했는데, NCCL (CUDA에서 GPU간 커뮤니케이션 관련 라이브러리인듯) 을 기본으로 사용한다고 한다. 그런데 NCCL 설정에 문제가 있으면 이런 현상이 발생한다고 한다. mirrored_strategy = tf.distribute.MirroredStrategy() 코드 실행하기 전에 Shell에서 다음과 같이 설정을 바꾸어주니 돌아가는 것 같다.export NCCL_P2P_DISABLE="1"export NCCL_IB_DISABLE=.. Low Power Computer Vision 후기 / 1위팀 코드 리뷰 다른 연구실 홈페이지를 구경하다보니, LPCV (Low Power Computer Vision) 이라는 경진대회에 참여했길래 관심이 생겨 이번에 나도 도전해 보게 되었다. 간단하게 얘기하면, 저전력으로 최대한 정확하게 동작하는 컴퓨터 비전 알고리즘을 찾는 경진대회라고 보면 된다. 매년 세부적인 주제는 바뀌는데, 올해(2023년)에는 Jetson Nano (2GB)정도의 저사양 장비에서 드론으로 촬영한 재난 장면을 Semantic segmentation하는 주제로 진행되었다. 샘플 이미지 하나만 보면 아래와 같다. 대회와 관련한 자세한 내용은 아래를 참고하면 된다. LPCV - Low Power Computer Vision LPCV - Low Power Computer Vision lpcv.ai 데이터는 .. 이전 1 2 3 4 5 6 7 8 ··· 11 다음
