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())
|