deepmd.pd.train.training

deepmd.pd.train.training#

Attributes#

log

Classes#

Trainer

Functions#

get_additional_data_requirement(_model)

whether_hessian(loss_params)

get_loss(loss_params, start_lr, _ntypes, _model)

get_single_model(_model_params)

get_model_for_wrapper(_model_params[, resuming, ...])

get_case_embd_config(_model_params)

model_change_out_bias(_model, _sample_func[, ...])

Module Contents#

deepmd.pd.train.training.log[source]#
class deepmd.pd.train.training.Trainer(config: dict[str, Any], training_data, stat_file_path=None, validation_data=None, init_model=None, restart_model=None, finetune_model=None, force_load=False, shared_links=None, finetune_links=None, init_frz_model=None)[source]#
restart_training[source]#
multi_task[source]#
finetune_update_stat = False[source]#
model_keys[source]#
rank[source]#
world_size[source]#
num_model[source]#
num_steps[source]#
disp_file[source]#
disp_freq[source]#
save_ckpt[source]#
save_freq[source]#
max_ckpt_keep[source]#
display_in_training[source]#
timing_in_training[source]#
change_bias_after_training[source]#
lcurve_should_print_header = True[source]#
model[source]#
warmup_steps[source]#
gradient_max_norm[source]#
wrapper[source]#
start_step = 0[source]#
enable_tensorboard[source]#
tensorboard_log_dir[source]#
tensorboard_freq[source]#
enable_profiler[source]#
profiling[source]#
profiling_file[source]#
run() → None[source]#
save_model(save_path, lr=0.0, step=0) → None[source]#
get_data(is_train=True, task_key='Default')[source]#
print_header(fout, train_results, valid_results) → None[source]#
print_on_training(fout, step_id, cur_lr, train_results, valid_results) → None[source]#
deepmd.pd.train.training.get_additional_data_requirement(_model)[source]#
deepmd.pd.train.training.whether_hessian(loss_params)[source]#
deepmd.pd.train.training.get_loss(loss_params, start_lr, _ntypes, _model)[source]#
deepmd.pd.train.training.get_single_model(_model_params)[source]#
deepmd.pd.train.training.get_model_for_wrapper(_model_params, resuming=False, _loss_params=None)[source]#
deepmd.pd.train.training.get_case_embd_config(_model_params)[source]#
deepmd.pd.train.training.model_change_out_bias(_model, _sample_func, _bias_adjust_mode='change-by-statistic')[source]#
Contents