wandb에서 image와 loss값들을 logging하는 코드를 추가하였습니다.
일단은 unlabel data의 hazy image와 이거를 clahe해서 임시gt? 만들어놓은것, 그리고 backbone에서 나온 원래의 output과 finetuning을 거친 output을 logging하였습니다.
loss값은 DC loss, bc loss, clahe loss, rec_loss, lwf loss, lwf_loss_label, lwf_loss_unlabel, total loss 등을 logging하도록 하였습니다.
로깅하고자 하는 것이 추가로 있으시다면 자유롭게 추가하시면 될 것 같습니다.
Changes
To reviewers
에폭 기준으로 로깅되도록 commit을 설정하였습니다. 즉, 매 에포크가 끝나면 logging화면이 띄워집니다. 이 부분을 수정하고 싶으시다면 wandb logging에서 commit=True할 곳을 로깅이 되었으면 하는 시점에 놓으시면 됩니다.
Dongwoo-Im
commented
2 years ago
What is this PR?
wandb에서 image와 loss값들을 logging하는 코드를 추가하였습니다. 일단은 unlabel data의 hazy image와 이거를 clahe해서 임시gt? 만들어놓은것, 그리고 backbone에서 나온 원래의 output과 finetuning을 거친 output을 logging하였습니다. loss값은 DC loss, bc loss, clahe loss, rec_loss, lwf loss, lwf_loss_label, lwf_loss_unlabel, total loss 등을 logging하도록 하였습니다. 로깅하고자 하는 것이 추가로 있으시다면 자유롭게 추가하시면 될 것 같습니다.
Changes
To reviewers
에폭 기준으로 로깅되도록 commit을 설정하였습니다. 즉, 매 에포크가 끝나면 logging화면이 띄워집니다. 이 부분을 수정하고 싶으시다면 wandb logging에서 commit=True할 곳을 로깅이 되었으면 하는 시점에 놓으시면 됩니다.