facebookresearch / odin

A simple and effective method for detecting out-of-distribution images in neural networks.
Other
526 stars 102 forks source link

Indexing first sample in batch? #19

Closed flpgrz closed 2 years ago

flpgrz commented 2 years ago

Hi,

https://github.com/facebookresearch/odin/blob/64e97962ccaed1fe979f43a089c0feb4d8b002fd/code/calData.py#L49

I don't get this line - is nnOutputs[0] the first sample in the batch?

flpgrz commented 2 years ago

I figured it out: batch size is 1.