Spaces:
Running
on
Zero
Running
on
Zero
derektan
commited on
Commit
·
996aee8
1
Parent(s):
23d27a1
Updated requirements.txt
Browse files- test_parameter.py +1 -1
test_parameter.py
CHANGED
|
@@ -162,7 +162,7 @@ INPUT_DIM = 4
|
|
| 162 |
EMBEDDING_DIM = 128
|
| 163 |
K_SIZE = 8 # 8
|
| 164 |
|
| 165 |
-
USE_GPU =
|
| 166 |
NUM_GPU = getenv("NUM_GPU", default=2, cast_type=int) # the number of GPUs
|
| 167 |
NUM_META_AGENT = getenv("NUM_META_AGENT", default=4, cast_type=int) # the number of processes
|
| 168 |
FOLDER_NAME = 'inference'
|
|
|
|
| 162 |
EMBEDDING_DIM = 128
|
| 163 |
K_SIZE = 8 # 8
|
| 164 |
|
| 165 |
+
USE_GPU = False # do you want to use GPUS?
|
| 166 |
NUM_GPU = getenv("NUM_GPU", default=2, cast_type=int) # the number of GPUs
|
| 167 |
NUM_META_AGENT = getenv("NUM_META_AGENT", default=4, cast_type=int) # the number of processes
|
| 168 |
FOLDER_NAME = 'inference'
|