m3gnet.config module

Data types

class m3gnet.config.DataType

Bases: object

Tensorflow and numpy data types. Used to choose between float16 and float32

np_float = 'float32'

np_int = 'int32'

classmethod set_dtype(data_type: str) → None

Class method to set the data types Args:

data_type (str): ‘16’ or ‘32’

tf_float = 'float32'

tf_int = 'int32'

m3gnet.config.set_global_dtypes(data_type) → None

Function to set the data types Args:

data_type (str): ‘16’ or ‘32’

Returns: