Spaces:
Running
on
A10G
Running
on
A10G
Update evaluate.py
Browse files- evaluate.py +9 -9
evaluate.py
CHANGED
@@ -254,21 +254,21 @@ def eval(test, standard, tmpdir):
|
|
254 |
return sum(scores) / len(scores), final_merged_intervals, comments
|
255 |
|
256 |
def install():
|
257 |
-
if torch.cuda.is_available():
|
258 |
-
|
259 |
-
|
260 |
-
else:
|
261 |
-
|
262 |
|
263 |
-
torch_version = torch.__version__.split('+')[0] # Get PyTorch version without build info
|
264 |
|
265 |
-
pip_command = f'pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html'
|
266 |
|
267 |
|
268 |
-
os.system(pip_command)
|
269 |
|
270 |
os.system('mim install mmengine')
|
271 |
-
|
272 |
os.system('mim install "mmdet>=3.1.0"')
|
273 |
os.system('git clone https://github.com/open-mmlab/mmpose.git')
|
274 |
os.chdir('mmpose')
|
|
|
254 |
return sum(scores) / len(scores), final_merged_intervals, comments
|
255 |
|
256 |
def install():
|
257 |
+
# if torch.cuda.is_available():
|
258 |
+
# cu_version = torch.version.cuda
|
259 |
+
# cu_version = f"cu{cu_version.replace('.', '')}" # Format it as 'cuXX' (e.g., 'cu113')
|
260 |
+
# else:
|
261 |
+
# cu_version = "cpu" # Fallback to CPU if no CUDA is available
|
262 |
|
263 |
+
# torch_version = torch.__version__.split('+')[0] # Get PyTorch version without build info
|
264 |
|
265 |
+
# pip_command = f'pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html'
|
266 |
|
267 |
|
268 |
+
# os.system(pip_command)
|
269 |
|
270 |
os.system('mim install mmengine')
|
271 |
+
os.system('mim install "mmcv>=2.0.1"')
|
272 |
os.system('mim install "mmdet>=3.1.0"')
|
273 |
os.system('git clone https://github.com/open-mmlab/mmpose.git')
|
274 |
os.chdir('mmpose')
|