feat: modify smp with amp

[MyeongheonChoi]

@wooyeolBaek 이 만든 코드에서 조금씩 수정했습니다. 자세한 것은 commit 내용에서 보면 될 것 같고, 꼭 알아둬야 할 대략적인 변경 사항은 다음과 같습니다.

1. mixed precision 적용
   • Unet + resnet34 기준 1epoch 2분 -> 1분 30초로 학습 시간 감소, memory 절반 정도로 감소
2. Unet, UnetPlusPlus 등 segmentation model을 parser에서 정하도록 수정
   • 이에 따라 model을 불러올 때 getattr을 이용해서 불러옴
3. wandb run name, 저장 되는 이름을 {segmentationmodel}{encoder}_{pretrained_weight}_시간 으로 수정
   • ex. Unet_resnet34_imagenet_221221_124500
4. criterion을 parser에서 정하도록 수정
   • 이에 따라 criterion을 불러올 때 getattr을 이용해서 불러옴

[MyeongheonChoi]

나중에 scheduler 같은 부분은 따로 .py 파일 만들어서 관리하는 방법을 찾아봐야 할 듯 합니다. parameter가 달라서 우선 어떻게 해야 할 지 못 정했어요.

[yyongjae]

확인했습니다 고마워용~

[wooyeolBaek]

@quasar529 , @KidsareBornStars 변경사항 확인 해주세요~

[KidsareBornStars]

확인했습니당 ㅎ