안녕하세요, 파이토치로 MaskRCNN을 공부하고 있는 사람입니다.
프로그래밍한 내역을 OS만 다른 동일한 조건(아래)의 환경에서 구동해보니, Linux는 정상구동되는데, 윈도우10은 이미지를 Tensor로 전환 후 forward 단계에서 아래와 같이 에러가 발생합니다. 예전에 이슈가 있었던거 같은데 해결방법이 확인되지 않아 경험하셨던분이 계시면 공유 부탁드립니다.
(forward 전단계까지는 Linux, 윈도우결과가 동일합니다..)
--- 구동환경 내역 ---
- OS : Linux (정상), Win10 (비정상)
- Torch : Libtorch 1.0.1 - CUDA100
- CUDA : 10.0
- CUDNN : 7.6.5
--- Source 부분 ----
std::tuple<torch::Tensor, torch::Tensor, torch::Tensor, torch::Tensor, torch::Tensor> FPNImpl::forward(at::Tensor x) {
x = c1_->forward(x);
...
---Torch Sourch 부분 ---
at::Tensor torch::nn::SequentialImpl::forward<at::Tensor, at::Tensor &> (at::Tensor &inputs)
--- 실행 후 에러메세지 ---
c1_->forward(x)
invalid argument 5: kernel size should be greater than zero, but got kH: 1 kW: -1103004587 at c:\a\w\1\s\windows\pytorch\aten\src\thcunn\generic/SpatialDilatedMaxPooling.cu:15