Spaces:
Running
Running
Commit
•
61855d2
1
Parent(s):
7f3745d
Update utils/voxelization.py
Browse files- utils/voxelization.py +1 -1
utils/voxelization.py
CHANGED
|
@@ -197,7 +197,7 @@ def processStructures(pdb_file, resids, clean=True):
|
|
| 197 |
|
| 198 |
results = [voxelize_single_notcentered(x) for x in environments]
|
| 199 |
|
| 200 |
-
voxels = torch.empty(len(results), 8, 32, 32, 32, device="
|
| 201 |
|
| 202 |
vox_env, prot_centers_list, prot_n_list, envs = zip(*results)
|
| 203 |
|
|
|
|
| 197 |
|
| 198 |
results = [voxelize_single_notcentered(x) for x in environments]
|
| 199 |
|
| 200 |
+
voxels = torch.empty(len(results), 8, 32, 32, 32, device="cpu")
|
| 201 |
|
| 202 |
vox_env, prot_centers_list, prot_n_list, envs = zip(*results)
|
| 203 |
|