huzey commited on
Commit
e77b20d
·
1 Parent(s): d48a41d

fix gpu memery

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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"