search

hunu12 / MALCOM

MALCOM : an OOD detector for CNN image classifiers. This is code for the paper "Convolutional Neural Networks with Compression Complexity Pooling for Out-of-Distribution Image Detection"
4 stars 0 forks source link

The error regards to the ImageNet size images #1

Open engkarat opened 4 years ago

engkarat commented 4 years ago

Hi,

Screenshot from 2020-11-02 23-06-05

Thank you for your code.

I got one question regards to the experiment that I am performing. Since your provided code is working well with the experimental dataset size, I apply it to the ImageNet size images (224x224) too. But I got the error shown in the above figure.

Could you please share with me how to solve this problem?

Best, Engkarat

hunu12 commented 4 years ago

Hi, I think the solution would be to

It seems to be happening from scalability issues. I am not sure if it would work on any machine other than mine. In fact, for convenience, I’ve tried to fully utilize the GPU resources like the code below.

https://github.com/hunu12/MALCOM/blob/ee36bf2b3224065df8d560d3a4d12559946bf378/ncd/cuda_utils.py#L34

I hope my answer is helpful. Anyway, I appreciate your interest for my work.

engkarat commented 4 years ago

Thank you very much for your reply. Let me try out. I will come back and update the result again.

Engkarat