dpgen.remote package#
Submodules#
dpgen.remote.decide_machine module#
- dpgen.remote.decide_machine.convert_mdata(mdata, task_types=['train', 'model_devi', 'fp'])[source]#
Convert mdata for DP-GEN main process. New convension is like mdata[“fp”][“machine”], DP-GEN needs mdata[“fp_machine”].
Notice that we deprecate the function which can automatically select one most avalaible machine, since this function was only used by Angus, and only supports for Slurm. In the future this can be implemented.
- Parameters:
- mdatadict
Machine parameters to be converted.
- task_typeslist of string
Type of tasks, default is [“train”, “model_devi”, “fp”]
- Returns:
- dict
mdata converted