deepmd.pt.utils.env

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]