replace bash script to docker-compose to build server

This commit is contained in:
2023-03-07 16:33:45 +08:00
parent ec19042d0d
commit 1f86146b12
7 changed files with 28 additions and 16 deletions

View File

@ -7,17 +7,18 @@ Try to use [MLflow](https://mlflow.org) platform to log PyTorch model training,
* MacOS 12.5
* Docker 20.10
# Dir
# Dirs
* **service**
* House MLflow service data, including MLflow artifacts, backend store and model registry
* **env**
* **mlflow.yaml**
* conda env yaml to run this repo
# Files
* **conda.yaml**
* conda env yaml to run this repo
* **start_mlflow_server.sh**
* a script to start MLflow server with basic configuration
* **docker-compose.yaml**
* a yaml to apply docker-compose to start MLflow service with basic configuration (run ```docker-compose -f docker-compose.yaml up```)
* **test_pytorch_m1.py**
* a script to test PyTorch on Apple M1 platform with GPU acceleration
* **train.py**