submit시 문제

[elzino]

안녕하세요 저는 지금 BaseDataset에서 getitem부분을 수정해서 사용하고 있는데요 제가 학습시킬때는 evaluation까지 잘 작동하는데 지금 학습시킨 모델을 submit하려고 하니까

input.size(-1) must be equal to input_size. Expected 2048, got 4128

라고 오류가 뜨면서 세션이 죽어버리네요 제 추측으로는 getitem부분이 naver의 코드대로 작동이되어서 저렇게 나오는 거 같은데 혹시 submit을 했을때 내부적으로 어떻게 작동하는것인가요? main함수를 따라서 실행되는게 아닌가요?

[ClovaAIAdmin]

main.py 의 infer() 함수에도 수정하신 getitem()을 반영하셨는지 확인 부탁드립니다. 감사합니다.

    def infer(wav_path):
        model.eval()
        device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

        input = get_spectrogram_feature(wav_path).unsqueeze(0)
        input = input.to(device)

        logit = model(input_variable=input, input_lengths=None, teacher_forcing_ratio=0)
        logit = torch.stack(logit, dim=1).to(device)

        y_hat = logit.max(-1)[1]
        hyp = label_to_string(y_hat)

        return hyp[0]

[elzino]

답변 감사합니다!