wi-lab
/

lwm

Feature Extraction

wireless-communication

few-shot-learning

Model card Files Files and versions

Sadjad Alikhani commited on Sep 16, 2024

Commit

7d37a39

·

verified ·

1 Parent(s): 6148f7d

Update inference.py

Files changed (1) hide show

inference.py +1 -1

inference.py CHANGED Viewed

@@ -26,7 +26,7 @@ warnings.filterwarnings('ignore')
 from input_preprocess import *
 # Device configuration
-device = torch.device(cuda if torch.cuda.is_available() else "cpu")
 if torch.cuda.is_available():
     torch.cuda.empty_cache()

 from input_preprocess import *
 # Device configuration
+device = torch.device('cuda' if torch.cuda.is_available() else "cpu")
 if torch.cuda.is_available():
     torch.cuda.empty_cache()