Spaces:
Running
on
Zero
Running
on
Zero
count the run
Browse files
app.py
CHANGED
@@ -7,6 +7,7 @@ from functools import partial
|
|
7 |
from io import BytesIO
|
8 |
import json
|
9 |
import os
|
|
|
10 |
import uuid
|
11 |
import zipfile
|
12 |
|
@@ -41,6 +42,7 @@ from ncut_pytorch.backbone import MODEL_DICT, LAYER_DICT, RES_DICT
|
|
41 |
from ncut_pytorch import NCUT
|
42 |
from ncut_pytorch import eigenvector_to_rgb, rotate_rgb_cube
|
43 |
|
|
|
44 |
|
45 |
DATASETS = {
|
46 |
'Common': [
|
@@ -1236,7 +1238,10 @@ def run_fn(
|
|
1236 |
"alignedcut_eig_norm_plot": alignedcut_eig_norm_plot,
|
1237 |
"advanced": advanced,
|
1238 |
}
|
1239 |
-
|
|
|
|
|
|
|
1240 |
|
1241 |
try:
|
1242 |
# try to aquiare GPU, can fail if the user is out of GPU quota
|
|
|
7 |
from io import BytesIO
|
8 |
import json
|
9 |
import os
|
10 |
+
from pprint import pprint
|
11 |
import uuid
|
12 |
import zipfile
|
13 |
|
|
|
42 |
from ncut_pytorch import NCUT
|
43 |
from ncut_pytorch import eigenvector_to_rgb, rotate_rgb_cube
|
44 |
|
45 |
+
RUN_COUNT = 0
|
46 |
|
47 |
DATASETS = {
|
48 |
'Common': [
|
|
|
1238 |
"alignedcut_eig_norm_plot": alignedcut_eig_norm_plot,
|
1239 |
"advanced": advanced,
|
1240 |
}
|
1241 |
+
global RUN_COUNT
|
1242 |
+
RUN_COUNT += 1
|
1243 |
+
print(f"Run Count: {RUN_COUNT}")
|
1244 |
+
pprint(kwargs)
|
1245 |
|
1246 |
try:
|
1247 |
# try to aquiare GPU, can fail if the user is out of GPU quota
|