Closed elzino closed 5 years ago
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]
답변 감사합니다!
안녕하세요 저는 지금 BaseDataset에서 getitem부분을 수정해서 사용하고 있는데요 제가 학습시킬때는 evaluation까지 잘 작동하는데 지금 학습시킨 모델을 submit하려고 하니까
라고 오류가 뜨면서 세션이 죽어버리네요 제 추측으로는 getitem부분이 naver의 코드대로 작동이되어서 저렇게 나오는 거 같은데 혹시 submit을 했을때 내부적으로 어떻게 작동하는것인가요? main함수를 따라서 실행되는게 아닌가요?