커뮤니티 대회 교육

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

[public: 0.86392] timm, pytorch-lightning

2023.01.08 16:06 1,556 조회 language

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

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

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

코드
DATA MONK
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분 정도 걸렸습니다.

로그인이 필요합니다
0 / 1000