Having the run function return the loss from the training.  Tweaking how the input dimension is computed if there is no molecular autoencoder provided.
Having the run function return the loss from the training.  Tweaking
how the input dimension is computed if there is no molecular
autoencoder provided.

ECP-CANDLE/Benchmarks,ECP-CANDLE/Benchmarks,ECP-CANDLE/Benchmarks