jhjacobsen / invertible-resnet

Official Code for Invertible Residual Networks

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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.