Dongwoo-Im
commented
2 years ago 확인했습니다 2가지 정도 수정해서 다시 push했는데요.
- utils.py load_model 에 type 인자를 받도록 해서, pretrain / finetune 구분이 가능하도록 했습니다. 올려주신 Epoch39.pth 이름을 PSD-Dehazeformer.pth 로 이름 바꾸어서 사용하시면 됩니다.
- finetune.py cr_loss 를 opt로 사용할지 말지 정할 수 있도록 했습니다.
확인하시면 merge해도 될 것 같습니다.
seonahmin
What is this PR?
동우님이 올려주셨던 코드에다가 dehazeformer까지 합쳐서 올려놨습니다! 일단 main(backbone학습시키는부분)이랑 finetune이 잘 돌아가는지 확인까지 한 상태입니다 나중에 사용하실 때 추가적인 오류가 있으면 말씀해주세요! 일단 이 코드는 finetune에도 label data에다가만 contrastive learning 하긴 했는데 아직 어떤 게 결과가 더 좋았는지 비교를 안 해본 상태라(바로 test만 돌리면 되어서 그건 바로 공유할게요!) 사용 안 하실거면 지우셔도 됩니다.
Changes
To reviewers
제가 모델 weight는 슬랙 댓글에다가 달아놓겠습니다! 깜빡하고 gitignore처리했네요,,