Spaces:
Running
on
Zero
Running
on
Zero
fix gpu memery
Browse files
app.py
CHANGED
@@ -167,6 +167,7 @@ def compute_ncut(
|
|
167 |
normalize_features=False,
|
168 |
indirect_connection=indirect_connection,
|
169 |
make_orthogonal=make_orthogonal,
|
|
|
170 |
).fit_transform(features.reshape(-1, features.shape[-1]))
|
171 |
# print(f"NCUT time: {time.time() - start:.2f}s")
|
172 |
logging_str += f"NCUT time: {time.time() - start:.2f}s\n"
|
|
|
167 |
normalize_features=False,
|
168 |
indirect_connection=indirect_connection,
|
169 |
make_orthogonal=make_orthogonal,
|
170 |
+
move_output_to_cpu=True,
|
171 |
).fit_transform(features.reshape(-1, features.shape[-1]))
|
172 |
# print(f"NCUT time: {time.time() - start:.2f}s")
|
173 |
logging_str += f"NCUT time: {time.time() - start:.2f}s\n"
|