search

FreyrS / dMaSIF

Other
193 stars 45 forks source link

a tiny error in save_protein_batch_single #4

Closed jiyanbio closed 3 years ago

jiyanbio commented 3 years ago

labels = P["labels"].view(-1, 1) if P1["labels"] is not None else 0.0 * predictions P1 --> P, isn't it?

jiyanbio commented 3 years ago

another issue: embedding = P["embedding_1"] --> embedding = P["embedding_1"] if pdb_idx == 1 else P["embedding_2"]

FreyrS commented 3 years ago

Hey @jiyanbio, you are absolutely correct, I've fixed these errors now, thanks a lot!

Freyr