deepmd.pt.train.training

Module Contents

Classes

Trainer

Functions

_model_change_out_bias(_model, new_type_map, ...)

Attributes

log

deepmd.pt.train.training.log[source]
class deepmd.pt.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]
run()[source]
save_model(save_path, lr=0.0, step=0)[source]
get_data(is_train=True, task_key='Default')[source]
print_header(fout, train_results, valid_results)[source]
print_on_training(fout, step_id, cur_lr, train_results, valid_results)[source]
deepmd.pt.train.training._model_change_out_bias(_model, new_type_map, _sample_func, _model_params)[source]