how to choose energy in inference stage

[Paulie-ai]

Hi, during the training of P2DFlow, the energy is precomputed and used as a label to perform cross-attention with the node embeddings. My question is: where does the energy come from during inference?
Thanks for your reply.

[BLEACH366]

Hi, as outlined in line 48 of ./experiments/utils.py and the paper, energy values are sampled within the range [0, 1] following a Boltzmann distribution. The goal is to ensure that the sampled protein conformations span a broad spectrum of energy levels.