치즈의 AI 녹이기

How can I process multi loss in pytorch? 본문

인공지능 대학원생의 생활/구글링

How can I process multi loss in pytorch?

개발자 치즈 2021. 6. 20. 23:35

한 개의 모델이 두 개 이상의 loss function을 갖고 있다면, pytorch에서는 어떻게 처리할까? 

 

#방법 1

loss1.backward()

loss2.backward()

loss3.backward()

optimizer.step()

 

#방법2

three loss = loss1+loss2+loss3

loss.backward()

optimizer.step()

 

참고 링크 : https://stackoverflow.com/questions/12453580/how-to-concatenate-items-in-a-list-to-a-single-string

'인공지능 대학원생의 생활 > 구글링' 카테고리의 다른 글

Batch normalization vs Layer normalization  (0) 2021.06.24
Weight decay  (0) 2021.06.24
Hard Negative Mining  (0) 2021.06.22
Teacher Forcing  (0) 2021.06.22
No module named 'apex'  (0) 2021.06.20