Spaces:

MohammedHamdy32
/

IDs_demo

Sleeping

MohammedHamdy32 commited on Jan 23, 2025

Commit

bc39774

verified ·

1 Parent(s): a03f138

chnage configration from cuda to cpu in config.py

Files changed (1) hide show

recognization/config.py CHANGED Viewed

@@ -6,7 +6,7 @@ class Config:
     output_channel = 32  # Adjust based on your architecture
     hidden_size = 256  # Adjust based on your architecture
     num_class = 182  # Number of output classes
-    device = 'cuda'  # or 'cuda' for GPU
     batch_max_length = 8  # Maximum sequence length for prediction
     # Adam optimizer
     adam = False

     output_channel = 32  # Adjust based on your architecture
     hidden_size = 256  # Adjust based on your architecture
     num_class = 182  # Number of output classes
+    device = 'cpu'  # or 'cuda' for GPU
     batch_max_length = 8  # Maximum sequence length for prediction
     # Adam optimizer
     adam = False