derektan commited on
Commit
996aee8
·
1 Parent(s): 23d27a1

Updated requirements.txt

Browse files
Files changed (1) hide show
  1. 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 = True # 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'
 
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'