diff --git a/benchmark/experimental/training_torch/README.md b/benchmark/experimental/training_torch/README.md new file mode 100644 index 00000000..684d0d73 --- /dev/null +++ b/benchmark/experimental/training_torch/README.md @@ -0,0 +1,2 @@ +# Training in PyTorch +This subdirectory contains code to train (some of) the benchmark models in PyTorch. As of October 9, 2023, it has not been verified to have the same results and computations as the reference model. In order to use this code towards a closed-division submission, you should load the weights from the reference model and verify that the computations and results are the same as the reference model. diff --git a/benchmark/training_torch/image_classification/README.md b/benchmark/experimental/training_torch/image_classification/README.md similarity index 100% rename from benchmark/training_torch/image_classification/README.md rename to benchmark/experimental/training_torch/image_classification/README.md diff --git a/benchmark/training_torch/image_classification/__init__.py b/benchmark/experimental/training_torch/image_classification/__init__.py similarity index 100% rename from benchmark/training_torch/image_classification/__init__.py rename to benchmark/experimental/training_torch/image_classification/__init__.py diff --git a/benchmark/training_torch/image_classification/download_cifar10_train_resnet.sh b/benchmark/experimental/training_torch/image_classification/download_cifar10_train_resnet.sh similarity index 100% rename from benchmark/training_torch/image_classification/download_cifar10_train_resnet.sh rename to benchmark/experimental/training_torch/image_classification/download_cifar10_train_resnet.sh diff --git a/benchmark/training_torch/image_classification/prepare_training_env.sh b/benchmark/experimental/training_torch/image_classification/prepare_training_env.sh similarity index 100% rename from benchmark/training_torch/image_classification/prepare_training_env.sh rename to benchmark/experimental/training_torch/image_classification/prepare_training_env.sh diff --git a/benchmark/training_torch/image_classification/requirements.txt b/benchmark/experimental/training_torch/image_classification/requirements.txt similarity index 100% rename from benchmark/training_torch/image_classification/requirements.txt rename to benchmark/experimental/training_torch/image_classification/requirements.txt diff --git a/benchmark/training_torch/image_classification/test.py b/benchmark/experimental/training_torch/image_classification/test.py similarity index 100% rename from benchmark/training_torch/image_classification/test.py rename to benchmark/experimental/training_torch/image_classification/test.py diff --git a/benchmark/training_torch/image_classification/train.py b/benchmark/experimental/training_torch/image_classification/train.py similarity index 100% rename from benchmark/training_torch/image_classification/train.py rename to benchmark/experimental/training_torch/image_classification/train.py diff --git a/benchmark/training_torch/image_classification/utils/__init__.py b/benchmark/experimental/training_torch/image_classification/utils/__init__.py similarity index 100% rename from benchmark/training_torch/image_classification/utils/__init__.py rename to benchmark/experimental/training_torch/image_classification/utils/__init__.py diff --git a/benchmark/training_torch/image_classification/utils/data.py b/benchmark/experimental/training_torch/image_classification/utils/data.py similarity index 100% rename from benchmark/training_torch/image_classification/utils/data.py rename to benchmark/experimental/training_torch/image_classification/utils/data.py diff --git a/benchmark/training_torch/image_classification/utils/model.py b/benchmark/experimental/training_torch/image_classification/utils/model.py similarity index 100% rename from benchmark/training_torch/image_classification/utils/model.py rename to benchmark/experimental/training_torch/image_classification/utils/model.py diff --git a/benchmark/training_torch/image_classification/utils/training.py b/benchmark/experimental/training_torch/image_classification/utils/training.py similarity index 100% rename from benchmark/training_torch/image_classification/utils/training.py rename to benchmark/experimental/training_torch/image_classification/utils/training.py