Tomasz Makowski comments

Repositories
Issues
Comments

Results 2 comments of


                                            Tomasz Makowski

Create documentation

Now the package is before first release but later we are planning to create documentation as a webpage. If you prefer to see high level code (now we don't have...

error: could not convert ‘at::CUDA(((c10::ScalarType)c10::kFloat))’ from ‘at::DeprecatedTypeProperties’ to ‘c10::IntArrayRef’ {aka ‘c10::ArrayRef<long int>’} 44 | auto max_val = at::zeros(torch::CUDA(at::kFloat), {batch, channel, width});

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));`