ziatdinovmax/atomai: v0.5.1

New functionalities:

• Add stochastic weights averaging option for both single model training and ensemble training
• Add SWAG-Gaussian method as one of the ensemble learning strategies

Breaking changes:

• The default output of atomnet.predictor.decode() and atomnet.predictor.run() is a raw prediction and a tuple with raw prediction and coordinates, respectively. The image data used as an input into a neural network is not returned by default anymore.