diff --git a/KNOWN_ISSUES.txt b/KNOWN_ISSUES.txt index cf930d7d2..cb9d3d2bb 100644 --- a/KNOWN_ISSUES.txt +++ b/KNOWN_ISSUES.txt @@ -2,4 +2,5 @@ KNOWN ISSUES for ai8x-training with PyToch 2.3.1 ================================================ * As of PyTorch 2.3.1, torch.compile() requires CUDA. - +* When running on Windows, train.py may require the switch "--compiler-mode none" if there is "no working triton installation". + See also https://github.com/pytorch/pytorch/issues/122094 diff --git a/README.md b/README.md index cf18db5e2..4df6c395d 100644 --- a/README.md +++ b/README.md @@ -1,6 +1,6 @@ # ADI MAX78000/MAX78002 Model Training and Synthesis -July 5, 2024 +July 22, 2024 **Note: This branch requires PyTorch 2. Please see the archive-1.8 branch for PyTorch 1.8 support. [KNOWN_ISSUES](KNOWN_ISSUES.txt) contains a list of known issues.** @@ -421,7 +421,7 @@ For minor updates, pull the latest code and install the updated wheels: (ai8x-training) $ git pull (ai8x-training) $ git submodule update --init (ai8x-training) $ pip3 install -U pip setuptools -(ai8x-training) $ pip3 install -U -r requirements.txt # or requirements-xxx.txt, as shown above +(ai8x-training) $ pip3 install -U -r requirements.txt # add --extra-index-url if needed, as shown above ``` ##### MSDK Updates diff --git a/README.pdf b/README.pdf index efa543414..3eca99513 100644 Binary files a/README.pdf and b/README.pdf differ diff --git a/distiller b/distiller index ea011e9ae..e89dc1362 160000 --- a/distiller +++ b/distiller @@ -1 +1 @@ -Subproject commit ea011e9ae4a68ae63d2d209517647d2d12ccacd9 +Subproject commit e89dc136276e307b8720dcef6b27073bc7f8bb74 diff --git a/pyproject.toml b/pyproject.toml index f048818f0..8aca08a13 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,5 +1,5 @@ [build-system] -requires = ["pip", "setuptools", "wheel"] +requires = ["pip", "setuptools >= 64", "wheel"] [project] name = "ai8x-training" diff --git a/requirements.txt b/requirements.txt index 824f5576f..a972c6c72 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,7 +4,7 @@ torchaudio==2.3.1 GitPython==3.1.43 Pillow==10.4.0 PyYAML==6.0.1 -albumentations==1.4.10 +albumentations==1.4.12 faiss-cpu==1.8.0.post1 batch-face==1.4.0 h5py==3.11.0 @@ -13,7 +13,7 @@ numpy==1.26.4 opencv-python==4.10.0.84 pycocotools==2.0.8 pyffmpeg==2.4.2.18.1 -pytorch-metric-learning==2.5.0 +pytorch-metric-learning==2.6.0 pytube==15.0.0 qrcode==7.4.2 scipy==1.14.0 @@ -22,4 +22,4 @@ tensorboard==2.17.0 tk==0.1.0 torchmetrics==1.4.0.post0 torchnet==0.0.4 --e distiller +-e distiller --config-settings editable_mode=strict