Add batch size to GraphModel predict

materialsvirtuallab / megnet

Graph Networks as a Universal Machine Learning Framework for Molecules and Crystals

BSD 3-Clause "New" or "Revised" License

497 stars 155 forks source link

Closed dgaines2 closed 2 years ago

dgaines2 commented 2 years ago

Allow predict_structures() and predict_graphs() to take an argument for batch_size