deepmd.pt_expt.utils.env

Contents

deepmd.pt_expt.utils.env#

Attributes#

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

Module Contents#

deepmd.pt_expt.utils.env.GLOBAL_NP_FLOAT_PRECISION[source]#

deepmd.pt_expt.utils.env.SAMPLER_RECORD[source]#

deepmd.pt_expt.utils.env.NUM_WORKERS[source]#

deepmd.pt_expt.utils.env.NUM_WORKERS = 0[source]#

deepmd.pt_expt.utils.env.LOCAL_RANK[source]#

deepmd.pt_expt.utils.env.LOCAL_RANK = 0[source]#

deepmd.pt_expt.utils.env.DEVICE[source]#

deepmd.pt_expt.utils.env.JIT = False[source]#

deepmd.pt_expt.utils.env.CACHE_PER_SYS = 5[source]#

deepmd.pt_expt.utils.env.ENERGY_BIAS_TRAINABLE = True[source]#

deepmd.pt_expt.utils.env.CUSTOM_OP_USE_JIT = False[source]#

deepmd.pt_expt.utils.env.PRECISION_DICT[source]#

deepmd.pt_expt.utils.env.GLOBAL_PT_FLOAT_PRECISION[source]#

deepmd.pt_expt.utils.env.GLOBAL_PT_ENER_FLOAT_PRECISION[source]#

deepmd.pt_expt.utils.env.RESERVED_PRECISION_DICT[source]#

deepmd.pt_expt.utils.env.DEFAULT_PRECISION = 'float64'[source]#