add gpu info
Browse files- app.py +0 -1
- 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())
|