DACON
search 구독 안내
알림 0

    MY XP

    XP 안내

    LEVEL 1

    다음 레벨까지 150 XP

    0/150 XP

    오늘의 XP

    0 / 800

    전체 XP

    0

    적립 XP

    0

    사용 XP

    0

    소셜 계정으로 로그인

    google
    구글 로그인
    아직 데이콘 계정이 없나요? 회원가입
    메일 인증 필요

    로그인 하시려면 아래 이메일로 인증이 필요합니다. 이메일을 다시 보내시겠습니까?
    [데이콘] 회원가입 인증메일

    데이콘 회원가입을 환영합니다. 메일 인증은 데이콘 회원가입을 위한 필수 절차입니다. 아래 이메일을 인증하여 회원가입 절차를 완료하여 주시기 바랍니다.

    이용약관

    개인정보 처리방침

    마케팅 정보 수신 동의

    포디블록 구조 추출 AI 경진대회

    알고리즘 | 비전 | 분류 | Accuracy

    연습
    대회안내 데이터 코드 공유 토크 리더보드

    팀 구성 D-0

     

    [public: 0.86392] timm, pytorch-lightning

    빙수
    2023.01.08 16:06 4,153 조회 language
    comments 5

    timm, pytorch-lighting을 사용한 코드입니다.
    원래 하나의 파이썬 파일로 작성한걸 노트북에 붙여넣어서 이런 모양이 되었습니다.

    이 코드는 모델 파일의 크기가 100MB 정도의 작은 모델을 사용했습니다.
    배치사이즈 32에서 VRAM 사용량은 8GB가 조금 안됩니다.
    아마 모델을 큰 모델로 바꾸는것만으로도 점수가 더 올라가지 않을까 생각합니다.

    코드 맨 밑에는 학습한 모델을 불러오는 방법을 적어두었으니 도움이 되길 바랍니다. 감사합니다.

    코드
    댓글 5개
    로그인이 필요합니다
    0 / 1000
    도비콘
    2023.01.08 16:26

    감사합니다.🐰

    한성대학교_김태민
    2023.01.08 22:51

    모델 불러와서 따로 BCE로스걸어서 학습시켜서 Multi label한건가요? 

    빙수
    2023.01.08 23:25

    네

    data yummy
    2023.01.19 15:57

    파이토치 사용하니 학습속도가 너무 느리네요... 에폭당 보통 얼마나 걸리시나요?

    빙수
    2023.01.19 22:48

    코드에 사용된 maxvit_rmlp_tiny_rw_256모델로 RTX 3070에서 에포크당 5분, 총 50분 정도 걸렸습니다.

    📣 댓글 작성 창의 위치가 댓글 리스트 상단으로 이동했습니다!
    to_list 목록으로
    이전 글
    연구 논문 "Perceiver"의 기준 모델
    대회 - 포디블록 구조 추출 AI 경진대회
    좋아요 9
    조회 3,133
    댓글 0
    2년 전
    현재 글
    [public: 0.86392] timm, pytorch-lightning
    대회 - 포디블록 구조 추출 AI 경진대회
    좋아요 10
    조회 4,153
    댓글 5
    2년 전
    다음 글
    [Baseline] Efficientnet B0을 활용한 Multi-Label Classification
    대회 - 포디블록 구조 추출 AI 경진대회
    좋아요 8
    조회 9,437
    댓글 14
    2년 전