JK.Min

Results 1 comments of JK.Min

> In all the places you have to change order of `at::zeros` arguments. For example, change `at::zeros(torch::CUDA(at::kFloat), {batch, channel, width});` into `at::zeros({batch, channel, width}, torch::CUDA(at::kFloat));` Thx!! You save my life!!