hookeai.miscellaneous.pruning.pruning_dataset.write_summary_file

write_summary_file(pruning_dir, pruning_params, mode='init', mode_data={})[source]

Write summary file.

Parameters:
  • pruning_dir (str) – Pruning main directory.

  • pruning_params (dict) – Pruning parameters.

  • mode ({'init', 'iter', 'end'}, default='init') – Summary mode.

  • mode_data (dict, default={}) – Summary mode data.