Open shawrby opened 5 years ago
y = x.item() # x는 텐서, y는 스칼라입니다.
로 했는데.. 로컬에선 문제 없는데, nsml 사용시 발생하네요. 혹시 해결 방안이 있나요?
원래 2차원 이상의 텐서는 .item()을 못쓰는 걸로 되어있습니다. nsml에서만 그런건 아닌거 같구요. 2차원 이상의 텐서에 대해서는 .tolist()를 쓰면 list로 반환되니 시도해보시길 바랍니다.
y = x.item() # x는 텐서, y는 스칼라입니다.
로 했는데.. 로컬에선 문제 없는데, nsml 사용시 발생하네요. 혹시 해결 방안이 있나요?