pub_soft/ScanNet
3
0
Fork 0
mirror of https://github.com/jertubiana/ScanNet.git synced 2026-06-04 13:44:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fixed (?) bug with nan due to numba x multiprocessing instability

Browse Source
This commit is contained in:
Jérôme Tubiana
2021-12-26 12:21:34 +02:00
parent cacd6b8320
commit fe5a172f52
5 changed files with 293 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
network/scannet.py
View File

@@ -773,9 +773,8 @@ if __name__ == '__main__':
import matplotlib
matplotlib.use('module://backend_interagg')
import matplotlib.pyplot as plt
import PDB_utils2
import Bio.PDB
import pipelines
from preprocessing import PDBio,pipelines
import numpy as np
with_atom = True
@@ -896,7 +895,6 @@ if __name__ == '__main__':
nrotations=nrotations,
dropout=dropout,
optimizer=optimizer, batch_size=1,
initial_values_folder='/specific/netapp5_2/iscb/wolfson/jeromet/Data/InterfacePrediction/initial_values/',
fresh_initial_values=True,
save_initial_values=False, n_init=1
)