코딩공부

Autoencoder를 만들면서 느낀것 본문

ML , DL (2019)/Pytorch

Autoencoder를 만들면서 느낀것

초보코더 2021. 1. 25. 11:15
반응형

- 임의의 데이터셋으로 dataset과 Dataloader를 만들면 ImageFolder가 제일 편하다  -> 추후 포스팅 예정

 

- 이미지에 AutoEncoder를 적용할 때 batch normalization을 안쓰면 gradient가 사라질 수 있다. ->꼭 써야함

 

- 큰 이미지에 대해서도 잘 돌아간다 -> 480 x 480 GrayScale 에 대해서 해본 결과 잘 동작

반응형