[SparseZoo v2 Bridge] Save IC training artifacts to ModelDirectory directory

[dbogunowicz]

Allows creating the ModelDirectory local directory from the direct output of the IC training procedure.

To test:

1. You need to install locally the sparsezoo's feature/ci-cd-refactor/validations branch
2. Then run:

import os.path
from sparsezoo.v2.model_directory import ModelDirectory
from sparseml.pytorch.image_classification.utils.helpers import get_model_directory

# where the training artifacts are
path_to_training_outputs = "/home/damian/sparseml/pytorch_vision/mobilenet_imagenette__01"
# where (potentially the logs are)
path_to_logs = "/home/damian/sparseml/src/sparseml/pytorch/image_classification/pytorch_vision_train/tensorboard logs/mobilenet_imagenette__01"
# where to save
save_dir = "/home/damian/my_model_dir"

# create model card if none exists
model_card_path = os.path.join(path_to_training_outputs, "model.md")
if not os.path.exists(model_card_path):
    open(model_card_path, "w").close()


get_model_directory(
    output_dir=save_dir,
    training_outputs_dir=path_to_training_outputs,
    logs_path=path_to_logs,
)

# print the resulting directory
start_path = save_dir
for path, dirs, files in os.walk(start_path):
    for filename in files:
        print(os.path.join(path, filename))

# this will fail in a controlled manner (the model card is empty)
assert ModelDirectory.from_directory(save_dir)

Output:

2022-07-01 10:39:44 sparseml.pytorch.image_classification.utils.helpers INFO     Created `ModelDirectory` folder locally in /home/damian/my_model_dir

/home/damian/my_model_dir/model.onnx
/home/damian/my_model_dir/model.md
/home/damian/my_model_dir/sample_inputs/inp-0002.npz
/home/damian/my_model_dir/sample_inputs/inp-0005.npz
...
/home/damian/my_model_dir/sample_outputs/out-0000.npz
/home/damian/my_model_dir/sample_outputs/out-0005.npz
...
/home/damian/my_model_dir/sample_labels/lab-0004.npz
/home/damian/my_model_dir/sample_labels/lab-0003.npz
...
/home/damian/my_model_dir/training/model.pth
/home/damian/my_model_dir/deployment/model.onnx
/home/damian/my_model_dir/logs/dummy_tf.log

ERROR:root:'NoneType' object has no attribute 'group'
Traceback (most recent call last):
  File "/home/damian/sparseml/src/sparseml/pytorch/image_classification/test.py", line 43, in <module>
    assert ModelDirectory.from_directory(save_dir)
  File "/home/damian/sparsezoo/src/sparsezoo/v2/model_directory.py", line 196, in from_directory
    return ModelDirectory(files=files, name="model_directory", path=directory_path)
  File "/home/damian/sparsezoo/src/sparsezoo/v2/model_directory.py", line 84, in __init__
    ] = self._sample_outputs_list_to_dict(
  File "/home/damian/sparsezoo/src/sparsezoo/v2/model_directory.py", line 474, in _sample_outputs_list_to_dict
    framework_name = self.model_card._validate_model_card()["framework"]
TypeError: 'NoneType' object is not subscriptable

[bfineran]

quick note - let's land this onto a separate feature branch as well and merge when sparsezoo refactor is ready to go