deepmd.pt.utils.env

Attributes

`GLOBAL_NP_FLOAT_PRECISION`
`SAMPLER_RECORD`
`NUM_WORKERS`
`LOCAL_RANK`
`LOCAL_RANK`
`DEVICE`
`JIT`
`CACHE_PER_SYS`
`ENERGY_BIAS_TRAINABLE`
`PRECISION_DICT`
`GLOBAL_PT_FLOAT_PRECISION`
`GLOBAL_PT_ENER_FLOAT_PRECISION`
`RESERVED_PRECISON_DICT`
`DEFAULT_PRECISION`

Module Contents

deepmd.pt.utils.env.GLOBAL_NP_FLOAT_PRECISION[source]

deepmd.pt.utils.env.SAMPLER_RECORD[source]

deepmd.pt.utils.env.NUM_WORKERS[source]

deepmd.pt.utils.env.LOCAL_RANK[source]

deepmd.pt.utils.env.LOCAL_RANK[source]

deepmd.pt.utils.env.DEVICE[source]

deepmd.pt.utils.env.JIT = False[source]

deepmd.pt.utils.env.CACHE_PER_SYS = 5[source]

deepmd.pt.utils.env.ENERGY_BIAS_TRAINABLE = True[source]

deepmd.pt.utils.env.PRECISION_DICT[source]

deepmd.pt.utils.env.GLOBAL_PT_FLOAT_PRECISION[source]

deepmd.pt.utils.env.GLOBAL_PT_ENER_FLOAT_PRECISION[source]

deepmd.pt.utils.env.RESERVED_PRECISON_DICT[source]

deepmd.pt.utils.env.DEFAULT_PRECISION = 'float64'[source]