최근 포스트

Pytorch 쓰면서 오류났던 점들 정리

1 분 소요

Model 설정할 때 return 값 확실하게 해두기! Crossentropyloss 사용 시 target의 shape은 [n,1]이면 안되고 [n, ]여야 한다!! -> input_size: [batch_size, class_size], output_size: [c...