torch.utils.model_dump.get_model_info

torch.utils.model_dump.get_model_info(path_or_file, title=None, extra_file_size_limit=16384) [исходный код]

Получить информацию о модели в формате, удобном для JSON.

Результат можно сохранить в файл model_info.json или передать в burn_in_info.