Exception training model

[peastman]

I tried to train a new aimnet model by following the instructions at https://isayevlab.github.io/aimnetcentral/train/. When I run the aimnet train command, it crashes with the exception

  File "/home/peastman/miniconda3/envs/aimnet2/lib/python3.12/site-packages/aimnet/train/utils.py", line 329, in log_loss_weights
    if loss_fn.weights is not None:
       ^^^^^^^^^^^^^^^
AttributeError: 'MTLoss' object has no attribute 'weights'

Looking at the definition of the MTLoss class, there is no attribute called weights, but log_loss_weights() expects it to exist.