Fix selection of device in evaluate.py

marco-rudolph / cs-flow

This is the official repository to the WACV 2022 paper "Fully Convolutional Cross-Scale-Flows for Image-based Defect Detection" by Marco Rudolph, Tom Wehrbein, Bodo Rosenhahn and Bastian Wandt.

111 stars 30 forks source link

Fix selection of device in evaluate.py #9

Closed AlessioGalluccio closed 2 years ago

AlessioGalluccio commented 2 years ago

Function evaluate could only be used in GPU setting. This minor fix permits to use it accordingly to the parameter "device" of config.py

marco-rudolph commented 2 years ago

Thank you!