a question about using large image to run classification
Christian-lyc opened this issue · comments
Hi, thank you so much for providing the code. I want to run it in my own dataset. I just set the batch size to 1 and even use 6 layers with 896*448 image, but still out of memory. Do you know how this error happens? Are there some ways to solve it? Thank you.
Did you switch off log determinant computation?
Thank you for your reply. I think yes. I set this: ignore_logdet=True.