Changes between Initial Version and Version 1 of Ticket #17555, comment 2


Ignore:
Timestamp:
May 8, 2025, 7:32:07 PM (6 months ago)
Author:
Tom Goddard

Legend:

Unmodified
Added
Removed
Modified
  • Ticket #17555, comment 2

    initial v1  
    1 Changing Boltz code to use bfloat16 weights and activations allowed successful predictions in cases that formerly gave out-of-memory errors for 9fz5 (1025 tokens) running in 181 seconds and 8sa0 (1371 residues) in 331 seconds.  Possibly it will be able to predict up to twice the former size limit of 900 residues on the Nvidia 4090 on minsky.  The structures looked fine but should be compared to the same predictions on the Mac Studio for accuracy.
     1Changing Boltz code to use bfloat16 weights and activations allowed successful predictions in cases that formerly gave out-of-memory errors for 9fz5 (1025 tokens) running in 181 seconds and 8sa0 (1371 residues) in 331 seconds.  Possibly it will be able to predict up to twice the former size limit of 900 residues on the Nvidia 4090 on minsky.  Unfortunately no, test from 1467 to 1770 residues all failed with out of memory. The structures looked fine but should be compared to the same predictions on the Mac Studio for accuracy.
    22
    33The Boltz code change in main.py in function predict() changes