Hi
I have ImageNet samples and did try to test your method on ImageNet but it seems that it causes out of memory. How did you address this issue?
MemoryError: Unable to allocate 450. GiB for an array with shape (401408, 224, 224, 3) and data type float64
Thanks