변이형 오토인코더에 MNIST 데이터 셋말고 다른 데이터셋을 넣으려고 합니다.

인공지능을 배운지 3개월된 학생입니다.
MNIST데이터 셋에 맞춰서 짜여진 변이형 오토인코더가 있습니다.
pytorch를 사용하고 있고, nn.linear을 사용하고 있습니다.

다른곳에서 가져온 코드 이기 때문에
입력값을 최대한 MNIST와 비슷하게 맞추었습니다.

MNIST는 2828의 값이지만, 우리는 220의 값으로 하려고 합니다…

입력값만 맞추고 숫자만 조정하면된다고 하는데,

MNIST일땐 잘 돌아가지만 인코더와 디코터의 출력값을 다르게 해야지 돌아갑니다…

그러다 loss를 계산할때 입출력값이 안맞다고 ㅠㅠ 에러 뜨는데 어떠게 해야할까요?

오류 메시지라도 알려주시면 좋을 것 같은데요.

안녕하세요, 올려주신 질문 글을 이해하지 못해서 답변이 다소 늦었습니다.

220의 값으로 하신다는 것은 어떤 말씀이실까요? 220x220 크기의 이미지일까요?

@Sung_Sue_Hwang 님께서 말씀하신 것처럼,
데이터에 대한 간단한 소개와 모델 코드, 에러 메시지를 함께 올려주시면 도움이 될 것 같습니다. :slight_smile: