Autoencoder를 만들면서 느낀것

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

 

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

 

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