why we need to move memory bank to CPU while inference time?

TooTouch / MemSeg

Unofficial re-implementation of MemSeg for Anomaly Detection

MIT License

184 stars 31 forks source link

why we need to move memory bank to CPU while inference time? #7

Closed prajyotmorey closed 1 year ago

prajyotmorey commented 1 year ago

Hii @TooTouch, In inference code, you have moved the memory bank to the CPU? is it necessary? Does it have any advantage to doing so? I want to run memsem on GPU, so what will be the best choice for it?

TooTouch commented 1 year ago

Hi, @prajyotmorey

If you want to use GPU, you can change device into gpu.

Best regards,