The highest accuracy is obtained with a rmax of 6 Å and 5 layer blocks. However, adopting a rmax of 5 Å provides a very close accuracy, with an increase of the MAE of the forces from 30.2 to 30.4 meV Å -1 and an increase of the MAE of the energy per atom from 0.130 to 0.164 meV. In addition, this choice saves about the 20% (22 hours) of the total GPU time of the training, which would results in a much more significative GPU time saving on the training on the whole data set, and will result in a more efficient model in production simulations as well.