pingnie commited on
Commit
659bce1
2 Parent(s): 034968f c3243aa

add gpu info

Browse files
Files changed (2) hide show
  1. app.py +0 -1
  2. backend-cli.py +1 -0
app.py CHANGED
@@ -1,5 +1,4 @@
1
  #!/usr/bin/env python
2
-
3
  import os
4
  import datetime
5
  import socket
 
1
  #!/usr/bin/env python
 
2
  import os
3
  import datetime
4
  import socket
backend-cli.py CHANGED
@@ -124,6 +124,7 @@ def request_to_result_name(request: EvalRequest) -> str:
124
 
125
 
126
  def process_evaluation(task: Task, eval_request: EvalRequest, limit: Optional[int] = None) -> dict:
 
127
  batch_size = eval_request.batch_size
128
 
129
  init_gpu_info = analyze_gpu_stats(parse_nvidia_smi())
 
124
 
125
 
126
  def process_evaluation(task: Task, eval_request: EvalRequest, limit: Optional[int] = None) -> dict:
127
+ batch_size = 1
128
  batch_size = eval_request.batch_size
129
 
130
  init_gpu_info = analyze_gpu_stats(parse_nvidia_smi())