search

f-lab-edu / virtual-try-on

4 stars 2 forks source link

config.yaml, cpu option #6

Closed seuno2 closed 8 months ago

seuno2 commented 8 months ago
  1. config.py > config.yaml 변경
  2. device=="cpu" 옵션에서 동작하도록 코드 수정
    • afwm.forward() 에서 correlation 연산 시, cupy kernel(C++) 내용을 numpy로 동작하도록 구현 중
seuno2 commented 8 months ago

correlation.py의 cupy_launch() 삭제 후 cpu 동작 가능 >> 추론 결과 이상해짐

7