Added 'device' to make switching between cuda and cpu possible

OpenAOI / anodet

Anomaly detection on images using features from pretrained neural networks.

MIT License

69 stars 22 forks source link

Open henrikBjorserud opened 2 years ago

henrikBjorserud commented 2 years ago

Added 'device' in two places to make switching between cuda and cpu possible