Hongmin

  • 홈
  • 태그
  • 방명록

flask 1

Subprocess - flask에 학습 기능 추가

pytorch에서 학습을 돌릴 때 data_loader는 num_workers의 값에 따라서 cpu core를 사용하게 됩니다. 요즘 사용하는 대부분의 컴퓨터는 코어가 여러개 달려있고, data load의 속도가 학습 속도에 상당히 영향을 많이 미치기 때문에 worker의 숫자를 잘 설정해야 합니다. flask에서 웹을 실행하면 app.run을 가동하는 프로세스 안에서 함수들이 실행되기 때문에 pytorch가 여러 개의 코어를 활성화 할 수 없습니다. 일반적인 model serving이 목적이신 분들은 bentoML을 사용하거나, ubuntu container 환경에서 gunicorn을 사용할 수도 있습니다. 그러나 저는 추론 기능은 필요 없고, 웹에 학습 기능만을 추가하려고 합니다. 때문에 이 글은 웹..

python 2022.01.20
1
더보기
프로필사진

일상블로그 blog.naver.com/nhmnhm0819 nhmnhm0819@naver.com

  • 분류 전체보기 (33)
    • ML theory (2)
    • CNN (Image Classification) (4)
    • Optical Flow (2)
    • GAN (5)
    • Detection (3)
    • python (2)
    • docker (3)
    • kubernetes (3)
    • computer (1)
    • MLOps (6)
    • Network (2)

Tag

감정인식, CNN, kubeflow, Variational Autoencoder, ASAPNet, DCLGAN, VAE, detectron, flownet, raft, Kubernetes, Detectron2, mask rcnn, NFNet, GAN, ESR-9, U-GAT-IT, gauGAN, subprocess, Optical Flow,

페이스북 트위터 플러그인

  • Facebook
  • Twitter

Archives

Calendar

«   2025/05   »
일 월 화 수 목 금 토
1 2 3
4 5 6 7 8 9 10
11 12 13 14 15 16 17
18 19 20 21 22 23 24
25 26 27 28 29 30 31

Copyright © Kakao Corp. All rights reserved.

티스토리툴바