Starting prediction...
/usr/local/lib/python3.7/site-packages/torch/jit/annotations.py:289: UserWarning: TorchScript will treat type annotations of Tensor dtype-specific subtypes as if they are normal Tensors. dtype constraints are not enforced in compilation either.
  warnings.warn("TorchScript will treat type annotations of Tensor "
start to load params /root/Uni-Fold-main/Alphafold/monomer.unifold.pt
start to predict T1024
{'aatype': torch.Size([1, 1, 408]), 'residue_index': torch.Size([1, 1, 408]), 'seq_length': torch.Size([1, 1]), 'template_aatype': torch.Size([1, 1, 4, 408]), 'template_all_atom_mask': torch.Size([1, 1, 4, 408, 37]), 'template_all_atom_positions': torch.Size([1, 1, 4, 408, 37, 3]), 'num_recycling_iters': torch.Size([1, 1]), 'is_distillation': torch.Size([8, 1]), 'seq_mask': torch.Size([1, 1, 408]), 'msa_mask': torch.Size([8, 1, 508, 408]), 'msa_row_mask': torch.Size([8, 1, 508]), 'template_mask': torch.Size([1, 1, 4]), 'template_pseudo_beta': torch.Size([1, 1, 4, 408, 3]), 'template_pseudo_beta_mask': torch.Size([1, 1, 4, 408]), 'template_torsion_angles_sin_cos': torch.Size([1, 1, 4, 408, 7, 2]), 'template_alt_torsion_angles_sin_cos': torch.Size([1, 1, 4, 408, 7, 2]), 'template_torsion_angles_mask': torch.Size([1, 1, 4, 408, 7]), 'residx_atom14_to_atom37': torch.Size([1, 1, 408, 14]), 'residx_atom37_to_atom14': torch.Size([1, 1, 408, 37]), 'atom14_atom_exists': torch.Size([1, 1, 408, 14]), 'atom37_atom_exists': torch.Size([1, 1, 408, 37]), 'target_feat': torch.Size([1, 1, 408, 22]), 'extra_msa': torch.Size([8, 1, 1024, 408]), 'extra_msa_mask': torch.Size([8, 1, 1024, 408]), 'extra_msa_row_mask': torch.Size([8, 1, 1024]), 'bert_mask': torch.Size([8, 1, 508, 408]), 'true_msa': torch.Size([8, 1, 508, 408]), 'extra_msa_has_deletion': torch.Size([8, 1, 1024, 408]), 'extra_msa_deletion_value': torch.Size([8, 1, 1024, 408]), 'msa_feat': torch.Size([8, 1, 508, 408, 49])}
Inference time: 138.50844680101727