├── .github ├── CODEOWNERS └── workflows │ └── cla.yml ├── .gitignore ├── CONTRIBUTING.md ├── LICENSE.md ├── README.md ├── emdenoise ├── .dockerignore ├── Dockerfile ├── README.md ├── fig1.jpg ├── fig2.jpg ├── main.py ├── mlcube.yaml ├── requirements.txt └── workspace │ └── parameters │ └── default.parameters.yaml ├── fets ├── README.md ├── data │ └── tmp.tar.gz ├── metrics │ ├── .gitignore │ ├── README.md │ ├── mlcube │ │ ├── mlcube.yaml │ │ └── workspace │ │ │ ├── data │ │ │ ├── ground_truth │ │ │ │ └── BraTS_example_seg.nii.gz │ │ │ └── predictions │ │ │ │ └── BraTS_example_seg.nii.gz │ │ │ └── parameters.yaml │ └── project │ │ ├── Dockerfile │ │ ├── metrics.py │ │ ├── mlcube.py │ │ └── requirements.txt ├── model │ ├── .gitignore │ ├── README.md │ ├── mlcube │ │ ├── mlcube.yaml │ │ └── workspace │ │ │ └── output │ │ │ ├── .dockerignore │ │ │ └── .gitignore │ └── project │ │ ├── Dockerfile │ │ ├── LICENSE │ │ ├── mlcube.py │ │ ├── model_ckpts │ │ └── example.ckpt │ │ ├── parameters.yaml │ │ ├── requirements.txt │ │ └── src │ │ ├── __init__.py │ │ ├── my_logic.py │ │ └── utils │ │ ├── __init__.py │ │ └── utilities.py └── preprocessing │ ├── .gitignore │ ├── README.md │ ├── mlcube │ ├── mlcube.yaml │ └── workspace │ │ ├── data │ │ └── BraTS_example_seg.nii.gz │ │ └── parameters.yaml │ └── project │ ├── Dockerfile │ ├── mlcube.py │ ├── preprocess.py │ ├── requirements.txt │ ├── run.sh │ ├── sanity_check.py │ └── statistics.py ├── getting-started ├── .gitignore ├── README.md ├── mlcube │ ├── mlcube.yaml │ └── workspace │ │ └── parameters.yaml └── project │ ├── 01_download_dataset.py │ ├── 02_preprocess_dataset.py │ ├── 03_train.py │ ├── Dockerfile │ └── requirements.txt ├── hello_world ├── .dockerignore ├── Dockerfile ├── README.md ├── hello_world.py ├── mlcube.yaml ├── requirements.txt └── workspace │ └── names │ └── alice.txt ├── matmul ├── .dockerignore ├── Dockerfile ├── README.md ├── matmul.py ├── matmul_constants.py ├── mlcube.yaml ├── requirements.txt └── workspace │ └── shapes.yaml ├── mnist ├── .dockerignore ├── Dockerfile ├── LICENSE ├── README.md ├── Singularity.recipe ├── mlcube.yaml ├── mnist.py ├── requirements.txt └── workspace │ ├── data.yaml │ └── train.yaml └── mnist_fl ├── .gitignore ├── LICENSE ├── pytorch ├── README.md ├── build │ ├── Dockerfile │ ├── mnist.py │ └── requirements.txt ├── mlcube.yaml └── workspace │ └── parameters │ └── default.parameters.yaml └── tensorflow ├── README.md ├── build ├── Dockerfile ├── mnist.py └── requirements.txt ├── mlcube.yaml └── workspace └── parameters └── default.parameters.yaml /.github/CODEOWNERS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/.github/CODEOWNERS -------------------------------------------------------------------------------- /.github/workflows/cla.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/.github/workflows/cla.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- 1 | .DS_Store 2 | .idea/ 3 | -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/README.md -------------------------------------------------------------------------------- /emdenoise/.dockerignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/emdenoise/.dockerignore -------------------------------------------------------------------------------- /emdenoise/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/emdenoise/Dockerfile -------------------------------------------------------------------------------- /emdenoise/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/emdenoise/README.md -------------------------------------------------------------------------------- /emdenoise/fig1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/emdenoise/fig1.jpg -------------------------------------------------------------------------------- /emdenoise/fig2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/emdenoise/fig2.jpg -------------------------------------------------------------------------------- /emdenoise/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/emdenoise/main.py -------------------------------------------------------------------------------- /emdenoise/mlcube.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/emdenoise/mlcube.yaml -------------------------------------------------------------------------------- /emdenoise/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/emdenoise/requirements.txt -------------------------------------------------------------------------------- /emdenoise/workspace/parameters/default.parameters.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/emdenoise/workspace/parameters/default.parameters.yaml -------------------------------------------------------------------------------- /fets/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/fets/README.md -------------------------------------------------------------------------------- /fets/data/tmp.tar.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/fets/data/tmp.tar.gz -------------------------------------------------------------------------------- /fets/metrics/.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__/ 2 | mlcube/workspace/results.yaml -------------------------------------------------------------------------------- /fets/metrics/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/fets/metrics/README.md -------------------------------------------------------------------------------- /fets/metrics/mlcube/mlcube.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/fets/metrics/mlcube/mlcube.yaml -------------------------------------------------------------------------------- /fets/metrics/mlcube/workspace/data/ground_truth/BraTS_example_seg.nii.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/fets/metrics/mlcube/workspace/data/ground_truth/BraTS_example_seg.nii.gz -------------------------------------------------------------------------------- /fets/metrics/mlcube/workspace/data/predictions/BraTS_example_seg.nii.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/fets/metrics/mlcube/workspace/data/predictions/BraTS_example_seg.nii.gz -------------------------------------------------------------------------------- /fets/metrics/mlcube/workspace/parameters.yaml: -------------------------------------------------------------------------------- 1 | treshold: 0.5 2 | eps: 0 -------------------------------------------------------------------------------- /fets/metrics/project/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/fets/metrics/project/Dockerfile -------------------------------------------------------------------------------- /fets/metrics/project/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/fets/metrics/project/metrics.py -------------------------------------------------------------------------------- /fets/metrics/project/mlcube.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/fets/metrics/project/mlcube.py -------------------------------------------------------------------------------- /fets/metrics/project/requirements.txt: -------------------------------------------------------------------------------- 1 | PyYAML 2 | typer 3 | numpy 4 | nibabel -------------------------------------------------------------------------------- /fets/model/.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__/ -------------------------------------------------------------------------------- /fets/model/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/fets/model/README.md -------------------------------------------------------------------------------- /fets/model/mlcube/mlcube.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/fets/model/mlcube/mlcube.yaml -------------------------------------------------------------------------------- /fets/model/mlcube/workspace/output/.dockerignore: -------------------------------------------------------------------------------- 1 | *.nii.gz 2 | -------------------------------------------------------------------------------- /fets/model/mlcube/workspace/output/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/fets/model/mlcube/workspace/output/.gitignore -------------------------------------------------------------------------------- /fets/model/project/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/fets/model/project/Dockerfile -------------------------------------------------------------------------------- /fets/model/project/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/fets/model/project/LICENSE -------------------------------------------------------------------------------- /fets/model/project/mlcube.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/fets/model/project/mlcube.py -------------------------------------------------------------------------------- /fets/model/project/model_ckpts/example.ckpt: -------------------------------------------------------------------------------- 1 | Model weights can be stored in this location -------------------------------------------------------------------------------- /fets/model/project/parameters.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/fets/model/project/parameters.yaml -------------------------------------------------------------------------------- /fets/model/project/requirements.txt: -------------------------------------------------------------------------------- 1 | PyYAML 2 | typer 3 | numpy 4 | SimpleITK -------------------------------------------------------------------------------- /fets/model/project/src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fets/model/project/src/my_logic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/fets/model/project/src/my_logic.py -------------------------------------------------------------------------------- /fets/model/project/src/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fets/model/project/src/utils/utilities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/fets/model/project/src/utils/utilities.py -------------------------------------------------------------------------------- /fets/preprocessing/.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__/ 2 | mlcube/workspace/results -------------------------------------------------------------------------------- /fets/preprocessing/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/fets/preprocessing/README.md -------------------------------------------------------------------------------- /fets/preprocessing/mlcube/mlcube.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/fets/preprocessing/mlcube/mlcube.yaml -------------------------------------------------------------------------------- /fets/preprocessing/mlcube/workspace/data/BraTS_example_seg.nii.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/fets/preprocessing/mlcube/workspace/data/BraTS_example_seg.nii.gz -------------------------------------------------------------------------------- /fets/preprocessing/mlcube/workspace/parameters.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/fets/preprocessing/mlcube/workspace/parameters.yaml -------------------------------------------------------------------------------- /fets/preprocessing/project/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/fets/preprocessing/project/Dockerfile -------------------------------------------------------------------------------- /fets/preprocessing/project/mlcube.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/fets/preprocessing/project/mlcube.py -------------------------------------------------------------------------------- /fets/preprocessing/project/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/fets/preprocessing/project/preprocess.py -------------------------------------------------------------------------------- /fets/preprocessing/project/requirements.txt: -------------------------------------------------------------------------------- 1 | PyYAML 2 | typer 3 | numpy 4 | nibabel 5 | tqdm 6 | SimpleITK -------------------------------------------------------------------------------- /fets/preprocessing/project/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/fets/preprocessing/project/run.sh -------------------------------------------------------------------------------- /fets/preprocessing/project/sanity_check.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/fets/preprocessing/project/sanity_check.py -------------------------------------------------------------------------------- /fets/preprocessing/project/statistics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/fets/preprocessing/project/statistics.py -------------------------------------------------------------------------------- /getting-started/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/getting-started/.gitignore -------------------------------------------------------------------------------- /getting-started/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/getting-started/README.md -------------------------------------------------------------------------------- /getting-started/mlcube/mlcube.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/getting-started/mlcube/mlcube.yaml -------------------------------------------------------------------------------- /getting-started/mlcube/workspace/parameters.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/getting-started/mlcube/workspace/parameters.yaml -------------------------------------------------------------------------------- /getting-started/project/01_download_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/getting-started/project/01_download_dataset.py -------------------------------------------------------------------------------- /getting-started/project/02_preprocess_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/getting-started/project/02_preprocess_dataset.py -------------------------------------------------------------------------------- /getting-started/project/03_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/getting-started/project/03_train.py -------------------------------------------------------------------------------- /getting-started/project/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/getting-started/project/Dockerfile -------------------------------------------------------------------------------- /getting-started/project/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/getting-started/project/requirements.txt -------------------------------------------------------------------------------- /hello_world/.dockerignore: -------------------------------------------------------------------------------- 1 | workspace/ 2 | -------------------------------------------------------------------------------- /hello_world/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/hello_world/Dockerfile -------------------------------------------------------------------------------- /hello_world/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/hello_world/README.md -------------------------------------------------------------------------------- /hello_world/hello_world.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/hello_world/hello_world.py -------------------------------------------------------------------------------- /hello_world/mlcube.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/hello_world/mlcube.yaml -------------------------------------------------------------------------------- /hello_world/requirements.txt: -------------------------------------------------------------------------------- 1 | typer -------------------------------------------------------------------------------- /hello_world/workspace/names/alice.txt: -------------------------------------------------------------------------------- 1 | Alice 2 | -------------------------------------------------------------------------------- /matmul/.dockerignore: -------------------------------------------------------------------------------- 1 | workspace/ 2 | -------------------------------------------------------------------------------- /matmul/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/matmul/Dockerfile -------------------------------------------------------------------------------- /matmul/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/matmul/README.md -------------------------------------------------------------------------------- /matmul/matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/matmul/matmul.py -------------------------------------------------------------------------------- /matmul/matmul_constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/matmul/matmul_constants.py -------------------------------------------------------------------------------- /matmul/mlcube.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/matmul/mlcube.yaml -------------------------------------------------------------------------------- /matmul/requirements.txt: -------------------------------------------------------------------------------- 1 | PyYAML==5.3 2 | tensorflow==2.1.0 3 | requests[security] -------------------------------------------------------------------------------- /matmul/workspace/shapes.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/matmul/workspace/shapes.yaml -------------------------------------------------------------------------------- /mnist/.dockerignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/mnist/.dockerignore -------------------------------------------------------------------------------- /mnist/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/mnist/Dockerfile -------------------------------------------------------------------------------- /mnist/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/mnist/LICENSE -------------------------------------------------------------------------------- /mnist/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/mnist/README.md -------------------------------------------------------------------------------- /mnist/Singularity.recipe: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/mnist/Singularity.recipe -------------------------------------------------------------------------------- /mnist/mlcube.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/mnist/mlcube.yaml -------------------------------------------------------------------------------- /mnist/mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/mnist/mnist.py -------------------------------------------------------------------------------- /mnist/requirements.txt: -------------------------------------------------------------------------------- 1 | PyYAML==5.3 2 | tensorflow==2.1.0 3 | requests[security] -------------------------------------------------------------------------------- /mnist/workspace/data.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/mnist/workspace/data.yaml -------------------------------------------------------------------------------- /mnist/workspace/train.yaml: -------------------------------------------------------------------------------- 1 | optimizer: "adam" 2 | train_epochs: 5 3 | batch_size: 32 4 | -------------------------------------------------------------------------------- /mnist_fl/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/mnist_fl/.gitignore -------------------------------------------------------------------------------- /mnist_fl/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/mnist_fl/LICENSE -------------------------------------------------------------------------------- /mnist_fl/pytorch/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/mnist_fl/pytorch/README.md -------------------------------------------------------------------------------- /mnist_fl/pytorch/build/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/mnist_fl/pytorch/build/Dockerfile -------------------------------------------------------------------------------- /mnist_fl/pytorch/build/mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/mnist_fl/pytorch/build/mnist.py -------------------------------------------------------------------------------- /mnist_fl/pytorch/build/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/mnist_fl/pytorch/build/requirements.txt -------------------------------------------------------------------------------- /mnist_fl/pytorch/mlcube.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/mnist_fl/pytorch/mlcube.yaml -------------------------------------------------------------------------------- /mnist_fl/pytorch/workspace/parameters/default.parameters.yaml: -------------------------------------------------------------------------------- 1 | epochs: 5 2 | batch_size: 32 3 | -------------------------------------------------------------------------------- /mnist_fl/tensorflow/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/mnist_fl/tensorflow/README.md -------------------------------------------------------------------------------- /mnist_fl/tensorflow/build/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/mnist_fl/tensorflow/build/Dockerfile -------------------------------------------------------------------------------- /mnist_fl/tensorflow/build/mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/mnist_fl/tensorflow/build/mnist.py -------------------------------------------------------------------------------- /mnist_fl/tensorflow/build/requirements.txt: -------------------------------------------------------------------------------- 1 | PyYAML==5.3 2 | tensorflow-cpu==2.1.0 3 | requests[security] -------------------------------------------------------------------------------- /mnist_fl/tensorflow/mlcube.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/mlcube_examples/HEAD/mnist_fl/tensorflow/mlcube.yaml -------------------------------------------------------------------------------- /mnist_fl/tensorflow/workspace/parameters/default.parameters.yaml: -------------------------------------------------------------------------------- 1 | optimizer: "adam" 2 | epochs: 5 3 | batch_size: 32 4 | --------------------------------------------------------------------------------