deepmd.nvnmd.fit package

nvnmd.fit =========.

Provides

1. continuous fitting network

2. quantized fitting network

Submodules

deepmd.nvnmd.fit.ener module

deepmd.nvnmd.fit.ener.one_layer_nvnmd(inputs, outputs_size, activation_fn=<function tanh>, precision=tf.float64, stddev=1.0, bavg=0.0, name='linear', reuse=None, seed=None, use_timestep=False, trainable=True, useBN=False, uniform_seed=False, initial_variables=None, mixed_prec=None, final_layer=False)

Build one layer with continuous or quantized value. Its weight and bias can be initialed with random or constant value.